Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / HEAD / . / src / backends / reference / RefLayerSupport.hpp
blob: 1b0f4a2bb590aea5d39e599f54c0be6408149873 [file] [log] [blame]
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1//
Tianle Cheng28288182024-02-23 17:56:54 +0000[diff] [blame]2// Copyright © 2017-2024 Arm Ltd and Contributors. All rights reserved.
David Beckecb56cd2018-09-05 12:52:57 +0100[diff] [blame]3// SPDX-License-Identifier: MIT
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]4//
5#pragma once
6
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]7#include <backendsCommon/LayerSupportBase.hpp>
arovir01a6824102018-08-28 17:40:45 +0100[diff] [blame]8
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]9namespace armnn
10{
11
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]12class RefLayerSupport : public LayerSupportBase
arovir014424b0a2018-10-04 10:46:04 +0100[diff] [blame]13{
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]14public:
Cathal Corbett34b429c2021-12-24 12:24:40 +0000[diff] [blame]15 bool IsLayerSupported(const LayerType& type,
16 const std::vector<TensorInfo>& infos,
17 const BaseDescriptor& descriptor,
18 const Optional<LstmInputParamsInfo>& lstmParamsInfo,
19 const Optional<QuantizedLstmInputParamsInfo>&,
20 Optional<std::string&> reasonIfUnsupported) const override;
21
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]22 bool IsActivationSupported(const TensorInfo& input,
23 const TensorInfo& output,
24 const ActivationDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]25 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]26
27 bool IsAdditionSupported(const TensorInfo& input0,
28 const TensorInfo& input1,
29 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]30 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]31
Nikhil Raj68c2c902019-09-19 11:21:11 +0100[diff] [blame]32 bool IsArgMinMaxSupported(const TensorInfo& input,
33 const TensorInfo& output,
34 const ArgMinMaxDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]35 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Nikhil Raj68c2c902019-09-19 11:21:11 +0100[diff] [blame]36
Samuel Yap6b478092022-07-06 15:36:03 +0100[diff] [blame]37 bool IsBatchMatMulSupported(const TensorInfo& inputX,
38 const TensorInfo& inputY,
39 const TensorInfo& output,
40 const BatchMatMulDescriptor& descriptor,
41 Optional<std::string &> reasonIfUnsupported = EmptyOptional()) const;
42
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]43 bool IsBatchNormalizationSupported(const TensorInfo& input,
44 const TensorInfo& output,
45 const TensorInfo& mean,
46 const TensorInfo& var,
47 const TensorInfo& beta,
48 const TensorInfo& gamma,
49 const BatchNormalizationDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]50 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]51
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000[diff] [blame]52 bool IsBatchToSpaceNdSupported(const TensorInfo& input,
53 const TensorInfo& output,
54 const BatchToSpaceNdDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]55 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000[diff] [blame]56
Idriss Chaouch98e383e2023-08-28 14:28:31 +0100[diff] [blame]57 bool IsBroadcastToSupported(const TensorInfo& input,
58 const TensorInfo& output,
59 const BroadcastToDescriptor& descriptor,
60 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
61
mathad01b392e982021-04-07 12:07:30 +0100[diff] [blame]62 bool IsCastSupported(const TensorInfo& input,
63 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]64 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
mathad01b392e982021-04-07 12:07:30 +0100[diff] [blame]65
Simon Obute51f67772021-09-03 15:50:13 +0100[diff] [blame]66 bool IsChannelShuffleSupported(const TensorInfo& input,
67 const TensorInfo& output,
68 const ChannelShuffleDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]69 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Simon Obute51f67772021-09-03 15:50:13 +0100[diff] [blame]70
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100[diff] [blame]71 bool IsComparisonSupported(const TensorInfo& input0,
72 const TensorInfo& input1,
73 const TensorInfo& output,
74 const ComparisonDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]75 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100[diff] [blame]76
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]77 bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,
78 const TensorInfo& output,
Cathal Corbett34b429c2021-12-24 12:24:40 +0000[diff] [blame]79 const OriginsDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]80 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]81
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]82 bool IsConstantSupported(const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]83 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]84
85 bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
86 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]87 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]88
Narumol Prangnawaratea54a012020-03-16 16:36:10 +0000[diff] [blame]89
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]90 bool IsConvertFp32ToFp16Supported(const TensorInfo& input,
91 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]92 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]93
94 bool IsConvolution2dSupported(const TensorInfo& input,
95 const TensorInfo& output,
96 const Convolution2dDescriptor& descriptor,
97 const TensorInfo& weights,
98 const Optional<TensorInfo>& biases,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]99 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]100
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]101 bool IsConvolution3dSupported(const TensorInfo& input,
102 const TensorInfo& output,
103 const Convolution3dDescriptor& descriptor,
104 const TensorInfo& weights,
105 const Optional<TensorInfo>& biases,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]106 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]107
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +0000[diff] [blame]108 bool IsDebugSupported(const TensorInfo& input,
109 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]110 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +0000[diff] [blame]111
Aron Virginas-Tar73f66422019-09-23 19:11:59 +0100[diff] [blame]112 bool IsDepthToSpaceSupported(const TensorInfo& input,
113 const TensorInfo& output,
114 const DepthToSpaceDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]115 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Aron Virginas-Tar73f66422019-09-23 19:11:59 +0100[diff] [blame]116
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]117 bool IsDepthwiseConvolutionSupported(const TensorInfo& input,
118 const TensorInfo& output,
119 const DepthwiseConvolution2dDescriptor& descriptor,
120 const TensorInfo& weights,
121 const Optional<TensorInfo>& biases,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]122 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]123
Nattapat Chaimanowong8a54ac02019-03-29 15:25:04 +0000[diff] [blame]124 bool IsDequantizeSupported(const TensorInfo& input,
125 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]126 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Nattapat Chaimanowong8a54ac02019-03-29 15:25:04 +0000[diff] [blame]127
Derek Lamberti6a5e5e82019-12-05 14:41:20 +0000[diff] [blame]128 bool IsDetectionPostProcessSupported(const TensorInfo& boxEncodings,
129 const TensorInfo& scores,
130 const TensorInfo& anchors,
131 const TensorInfo& detectionBoxes,
132 const TensorInfo& detectionClasses,
133 const TensorInfo& detectionScores,
134 const TensorInfo& numDetections,
Narumol Prangnawaratbc67cef2019-01-31 15:31:54 +0000[diff] [blame]135 const DetectionPostProcessDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]136 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Narumol Prangnawaratbc67cef2019-01-31 15:31:54 +0000[diff] [blame]137
Pablo Tellof0bd6832019-04-26 17:58:13 +0100[diff] [blame]138 bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo& input,
139 const TensorInfo& output,
140 const DepthwiseConvolution2dDescriptor& descriptor,
141 const TensorInfo& weights,
142 const Optional<TensorInfo>& biases,
143 Optional<std::string&> reasonIfUnsupported =
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]144 EmptyOptional()) const;
Pablo Tellof0bd6832019-04-26 17:58:13 +0100[diff] [blame]145
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]146 bool IsDivisionSupported(const TensorInfo& input0,
147 const TensorInfo& input1,
148 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]149 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]150
josh minor4a3c6102020-01-06 16:40:46 -0600[diff] [blame]151 bool IsElementwiseUnarySupported(const TensorInfo& input,
152 const TensorInfo& output,
153 const ElementwiseUnaryDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]154 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
josh minor4a3c6102020-01-06 16:40:46 -0600[diff] [blame]155
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]156 bool IsFakeQuantizationSupported(const TensorInfo& input,
157 const FakeQuantizationDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]158 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]159
Ryan OSheaf4bfa6a2020-06-10 11:33:37 +0100[diff] [blame]160 bool IsFillSupported(const TensorInfo& input,
161 const TensorInfo& output,
162 const FillDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]163 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Ryan OSheaf4bfa6a2020-06-10 11:33:37 +0100[diff] [blame]164
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]165 bool IsFloorSupported(const TensorInfo& input,
166 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]167 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]168
169 bool IsFullyConnectedSupported(const TensorInfo& input,
170 const TensorInfo& output,
171 const TensorInfo& weights,
172 const TensorInfo& biases,
173 const FullyConnectedDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]174 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]175
Teresa Charlinb2d3ec52022-04-12 22:07:09 +0100[diff] [blame]176 bool IsGatherNdSupported(const TensorInfo& input0,
177 const TensorInfo& input1,
178 const TensorInfo& output,
179 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
180
narpra014951d842019-01-18 16:53:53 +0000[diff] [blame]181 bool IsGatherSupported(const TensorInfo& input0,
182 const TensorInfo& input1,
183 const TensorInfo& output,
Teresa Charlin52664732020-06-29 16:27:03 +0100[diff] [blame]184 const GatherDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]185 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
narpra014951d842019-01-18 16:53:53 +0000[diff] [blame]186
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]187 bool IsInputSupported(const TensorInfo& input,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]188 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]189
Kevin May09ca49c2019-10-09 12:37:34 +0100[diff] [blame]190 bool IsInstanceNormalizationSupported(const TensorInfo& input,
191 const TensorInfo& output,
192 const InstanceNormalizationDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]193 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Kevin May09ca49c2019-10-09 12:37:34 +0100[diff] [blame]194
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]195 bool IsL2NormalizationSupported(const TensorInfo& input,
196 const TensorInfo& output,
197 const L2NormalizationDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]198 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]199
James Conroyaba90cd2020-11-06 16:28:18 +0000[diff] [blame]200 bool IsLogicalBinarySupported(const TensorInfo& input0,
201 const TensorInfo& input1,
202 const TensorInfo& output,
203 const LogicalBinaryDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]204 Optional<std::string&> reasonIfUnsupported) const;
James Conroyaba90cd2020-11-06 16:28:18 +0000[diff] [blame]205
Aron Virginas-Tare662a942019-10-14 15:12:00 +0100[diff] [blame]206 bool IsLogSoftmaxSupported(const TensorInfo& input,
207 const TensorInfo& output,
208 const LogSoftmaxDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]209 Optional<std::string&> reasonIfUnsupported) const;
Aron Virginas-Tare662a942019-10-14 15:12:00 +0100[diff] [blame]210
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]211 bool IsLstmSupported(const TensorInfo& input,
212 const TensorInfo& outputStateIn,
213 const TensorInfo& cellStateIn,
214 const TensorInfo& scratchBuffer,
215 const TensorInfo& outputStateOut,
216 const TensorInfo& cellStateOut,
217 const TensorInfo& output,
218 const LstmDescriptor& descriptor,
Jan Eilersd01a83c2019-07-03 18:20:40 +0100[diff] [blame]219 const LstmInputParamsInfo& paramsInfo,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]220 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]221
saoste012df12b32018-11-28 16:57:20 +0000[diff] [blame]222 bool IsMaximumSupported(const TensorInfo& input0,
223 const TensorInfo& input1,
224 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]225 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
saoste012df12b32018-11-28 16:57:20 +0000[diff] [blame]226
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]227 bool IsMeanSupported(const TensorInfo& input,
228 const TensorInfo& output,
229 const MeanDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]230 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]231
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]232 bool IsMemCopySupported(const TensorInfo& input,
233 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]234 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]235
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000[diff] [blame]236 bool IsMinimumSupported(const TensorInfo& input0,
237 const TensorInfo& input1,
238 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]239 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000[diff] [blame]240
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]241 bool IsMultiplicationSupported(const TensorInfo& input0,
242 const TensorInfo& input1,
243 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]244 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]245
246 bool IsNormalizationSupported(const TensorInfo& input,
247 const TensorInfo& output,
248 const NormalizationDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]249 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]250
251 bool IsOutputSupported(const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]252 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]253
254 bool IsPadSupported(const TensorInfo& input,
255 const TensorInfo& output,
256 const PadDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]257 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]258
259 bool IsPermuteSupported(const TensorInfo& input,
260 const TensorInfo& output,
261 const PermuteDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]262 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]263
264 bool IsPooling2dSupported(const TensorInfo& input,
265 const TensorInfo& output,
266 const Pooling2dDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]267 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]268
Tamás Nyíri7b885b32021-10-26 14:47:57 +0100[diff] [blame]269 bool IsPooling3dSupported(const TensorInfo& input,
270 const TensorInfo& output,
271 const Pooling3dDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]272 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Tamás Nyíri7b885b32021-10-26 14:47:57 +0100[diff] [blame]273
Derek Lamberti5f400d62019-03-25 15:41:58 +0000[diff] [blame]274 bool IsQuantizeSupported(const TensorInfo& input,
275 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]276 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Derek Lamberti5f400d62019-03-25 15:41:58 +0000[diff] [blame]277
James Conroy4f1f8992020-04-29 20:01:10 +0100[diff] [blame]278 bool IsQLstmSupported(const TensorInfo& input,
279 const TensorInfo& previousOutputIn,
280 const TensorInfo& previousCellStateIn,
281 const TensorInfo& outputStateOut,
282 const TensorInfo& cellStateOut,
283 const TensorInfo& output,
284 const QLstmDescriptor& descriptor,
285 const LstmInputParamsInfo& paramsInfo,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]286 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
James Conroy4f1f8992020-04-29 20:01:10 +0100[diff] [blame]287
Finn Williams2605b232020-06-10 15:53:46 +0100[diff] [blame]288 bool IsRankSupported(const TensorInfo& input,
289 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]290 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Finn Williams2605b232020-06-10 15:53:46 +0100[diff] [blame]291
Sadik Armagan0c3ea5b2021-02-03 09:29:30 +0000[diff] [blame]292 bool IsReduceSupported(const TensorInfo& input,
293 const TensorInfo& output,
294 const ReduceDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]295 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Sadik Armagan0c3ea5b2021-02-03 09:29:30 +0000[diff] [blame]296
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]297 bool IsReshapeSupported(const TensorInfo& input,
Kevin Maya023c402019-12-12 17:28:05 +0000[diff] [blame]298 const TensorInfo& output,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]299 const ReshapeDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]300 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]301
Teresa Charlin970f43b2019-07-01 13:51:07 +0100[diff] [blame]302 bool IsResizeSupported(const TensorInfo& input,
303 const TensorInfo& output,
304 const ResizeDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]305 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Keith Davis3ae3f972021-05-21 16:33:48 +0100[diff] [blame]306
Tracy Narinebb8d7592023-07-13 16:50:54 +0100[diff] [blame]307 bool IsReverseV2Supported(const TensorInfo& input0,
308 const TensorInfo& input1,
Tianle Cheng988354d2023-06-28 13:20:47 +0100[diff] [blame]309 const TensorInfo& output,
Tianle Cheng988354d2023-06-28 13:20:47 +0100[diff] [blame]310 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
311
Tianle Cheng28288182024-02-23 17:56:54 +0000[diff] [blame]312 bool IsScatterNdSupported(const TensorInfo& input,
313 const TensorInfo& indices,
314 const TensorInfo& updates,
315 const TensorInfo& output,
316 const ScatterNdDescriptor& descriptor,
317 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
318
Keith Davis3ae3f972021-05-21 16:33:48 +0100[diff] [blame]319 bool IsShapeSupported(const TensorInfo& input,
320 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]321 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Keith Davis3ae3f972021-05-21 16:33:48 +0100[diff] [blame]322
Aron Virginas-Tar92b9f872019-09-17 17:27:04 +0100[diff] [blame]323 bool IsSliceSupported(const TensorInfo& input,
324 const TensorInfo& output,
325 const SliceDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]326 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Aron Virginas-Tar92b9f872019-09-17 17:27:04 +0100[diff] [blame]327
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]328 bool IsSoftmaxSupported(const TensorInfo& input,
329 const TensorInfo& output,
330 const SoftmaxDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]331 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]332
Nattapat Chaimanowong3ea76d52018-11-09 14:10:38 +0000[diff] [blame]333 bool IsSpaceToBatchNdSupported(const TensorInfo& input,
334 const TensorInfo& output,
335 const SpaceToBatchNdDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]336 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Nattapat Chaimanowong3ea76d52018-11-09 14:10:38 +0000[diff] [blame]337
Keith Davisa57eccb2019-06-14 17:33:22 +0100[diff] [blame]338 bool IsSpaceToDepthSupported(const TensorInfo& input,
339 const TensorInfo& output,
340 const SpaceToDepthDescriptor& descriptor,
341 Optional<std::string&> reasonIfUnsupported = EmptyOptional())
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]342 const;
Keith Davisa57eccb2019-06-14 17:33:22 +0100[diff] [blame]343
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100[diff] [blame]344 bool IsSplitterSupported(const TensorInfo& input,
345 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
346 const ViewsDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]347 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100[diff] [blame]348
Matthew Jackson81e601c2019-07-11 12:07:09 +0100[diff] [blame]349 bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
350 const TensorInfo& output,
351 const StackDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]352 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Matthew Jackson81e601c2019-07-11 12:07:09 +0100[diff] [blame]353
Nattapat Chaimanowong1216b582018-11-23 15:33:41 +0000[diff] [blame]354 bool IsStridedSliceSupported(const TensorInfo& input,
355 const TensorInfo& output,
356 const StridedSliceDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]357 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Nattapat Chaimanowong1216b582018-11-23 15:33:41 +0000[diff] [blame]358
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]359 bool IsSubtractionSupported(const TensorInfo& input0,
360 const TensorInfo& input1,
361 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]362 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Matteo Martincighab9e5252019-06-13 17:27:46 +0100[diff] [blame]363
364 bool IsPreluSupported(const TensorInfo& input,
365 const TensorInfo& alpha,
366 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]367 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Aron Virginas-Tar98180ef2019-06-26 15:02:47 +0100[diff] [blame]368
Teresa Charlin79a06a52023-07-13 17:16:45 +0100[diff] [blame]369 bool IsTileSupported(const TensorInfo& input,
370 const TensorInfo& output,
371 const TileDescriptor& descriptor,
372 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
373
Aron Virginas-Tar98180ef2019-06-26 15:02:47 +0100[diff] [blame]374 bool IsTransposeConvolution2dSupported(
375 const TensorInfo& input,
376 const TensorInfo& output,
377 const TransposeConvolution2dDescriptor& descriptor,
378 const TensorInfo& weights,
379 const Optional<TensorInfo>& biases,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]380 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000[diff] [blame]381
382 bool IsTransposeSupported(const TensorInfo& input,
383 const TensorInfo& output,
384 const TransposeDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]385 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Narumol Prangnawarate5339e72021-07-28 17:33:28 +0100[diff] [blame]386
387 bool IsUnidirectionalSequenceLstmSupported(
388 const TensorInfo& input,
389 const TensorInfo& outputStateIn,
390 const TensorInfo& cellStateIn,
Mike Kelly12994962022-04-21 11:57:09 +0100[diff] [blame]391 const TensorInfo& outputStateOut,
392 const TensorInfo& cellStateOut,
Narumol Prangnawarate5339e72021-07-28 17:33:28 +0100[diff] [blame]393 const TensorInfo& output,
Narumol Prangnawarate5339e72021-07-28 17:33:28 +0100[diff] [blame]394 const UnidirectionalSequenceLstmDescriptor& descriptor,
395 const LstmInputParamsInfo& paramsInfo,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]396 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir014424b0a2018-10-04 10:46:04 +0100[diff] [blame]397};
398
Aron Virginas-Tarb5acbb72018-10-15 11:11:51 +0100[diff] [blame]399} // namespace armnn