blob: b64244db244488ae525541ff26e65e49e51133da [file] [log] [blame]
telsoa014fcda012018-03-09 14:13:49 +00001//
Teresa Charlin52664732020-06-29 16:27:03 +01002// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
David Beckecb56cd2018-09-05 12:52:57 +01003// SPDX-License-Identifier: MIT
telsoa014fcda012018-03-09 14:13:49 +00004//
5#pragma once
6
Matteo Martincigh992d6dc2019-01-10 17:34:20 +00007#include <backendsCommon/LayerSupportBase.hpp>
arovir01a6824102018-08-28 17:40:45 +01008
telsoa014fcda012018-03-09 14:13:49 +00009namespace armnn
10{
11
Matteo Martincigh992d6dc2019-01-10 17:34:20 +000012class RefLayerSupport : public LayerSupportBase
arovir014424b0a2018-10-04 10:46:04 +010013{
arovir011c7c81b2018-10-08 11:34:28 +010014public:
Cathal Corbett34b429c2021-12-24 12:24:40 +000015 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 +010022 bool IsActivationSupported(const TensorInfo& input,
23 const TensorInfo& output,
24 const ActivationDescriptor& descriptor,
25 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
26
27 bool IsAdditionSupported(const TensorInfo& input0,
28 const TensorInfo& input1,
29 const TensorInfo& output,
30 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
31
Nikhil Raj68c2c902019-09-19 11:21:11 +010032 bool IsArgMinMaxSupported(const TensorInfo& input,
33 const TensorInfo& output,
34 const ArgMinMaxDescriptor& descriptor,
35 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
36
Samuel Yap4b7a34d2022-07-06 15:36:03 +010037 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 +010043 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,
50 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
51
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +000052 bool IsBatchToSpaceNdSupported(const TensorInfo& input,
53 const TensorInfo& output,
54 const BatchToSpaceNdDescriptor& descriptor,
55 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
56
mathad01b392e982021-04-07 12:07:30 +010057 bool IsCastSupported(const TensorInfo& input,
58 const TensorInfo& output,
59 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
60
Simon Obute51f67772021-09-03 15:50:13 +010061 bool IsChannelShuffleSupported(const TensorInfo& input,
62 const TensorInfo& output,
63 const ChannelShuffleDescriptor& descriptor,
64 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
65
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +010066 bool IsComparisonSupported(const TensorInfo& input0,
67 const TensorInfo& input1,
68 const TensorInfo& output,
69 const ComparisonDescriptor& descriptor,
70 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
71
Jim Flynn906f9462019-05-10 13:55:21 +010072 bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,
73 const TensorInfo& output,
Cathal Corbett34b429c2021-12-24 12:24:40 +000074 const OriginsDescriptor& descriptor,
Jim Flynn906f9462019-05-10 13:55:21 +010075 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
76
arovir011c7c81b2018-10-08 11:34:28 +010077 bool IsConstantSupported(const TensorInfo& output,
78 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
79
Narumol Prangnawarat7ddbbae2020-03-13 10:26:05 +000080 bool IsConvertBf16ToFp32Supported(const TensorInfo& input,
81 const TensorInfo& output,
82 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
83
arovir011c7c81b2018-10-08 11:34:28 +010084 bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
85 const TensorInfo& output,
86 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
87
Narumol Prangnawaratea54a012020-03-16 16:36:10 +000088 bool IsConvertFp32ToBf16Supported(const TensorInfo& input,
89 const TensorInfo& output,
90 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
91
arovir011c7c81b2018-10-08 11:34:28 +010092 bool IsConvertFp32ToFp16Supported(const TensorInfo& input,
93 const TensorInfo& output,
94 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
95
96 bool IsConvolution2dSupported(const TensorInfo& input,
97 const TensorInfo& output,
98 const Convolution2dDescriptor& descriptor,
99 const TensorInfo& weights,
100 const Optional<TensorInfo>& biases,
101 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
102
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100103 bool IsConvolution3dSupported(const TensorInfo& input,
104 const TensorInfo& output,
105 const Convolution3dDescriptor& descriptor,
106 const TensorInfo& weights,
107 const Optional<TensorInfo>& biases,
108 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
109
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +0000110 bool IsDebugSupported(const TensorInfo& input,
111 const TensorInfo& output,
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +0000112 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
113
Aron Virginas-Tar73f66422019-09-23 19:11:59 +0100114 bool IsDepthToSpaceSupported(const TensorInfo& input,
115 const TensorInfo& output,
116 const DepthToSpaceDescriptor& descriptor,
117 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
118
arovir011c7c81b2018-10-08 11:34:28 +0100119 bool IsDepthwiseConvolutionSupported(const TensorInfo& input,
120 const TensorInfo& output,
121 const DepthwiseConvolution2dDescriptor& descriptor,
122 const TensorInfo& weights,
123 const Optional<TensorInfo>& biases,
124 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
125
Nattapat Chaimanowong8a54ac02019-03-29 15:25:04 +0000126 bool IsDequantizeSupported(const TensorInfo& input,
127 const TensorInfo& output,
128 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
129
Derek Lamberti6a5e5e82019-12-05 14:41:20 +0000130 bool IsDetectionPostProcessSupported(const TensorInfo& boxEncodings,
131 const TensorInfo& scores,
132 const TensorInfo& anchors,
133 const TensorInfo& detectionBoxes,
134 const TensorInfo& detectionClasses,
135 const TensorInfo& detectionScores,
136 const TensorInfo& numDetections,
Narumol Prangnawaratbc67cef2019-01-31 15:31:54 +0000137 const DetectionPostProcessDescriptor& descriptor,
138 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
139
Pablo Tellof0bd6832019-04-26 17:58:13 +0100140 bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo& input,
141 const TensorInfo& output,
142 const DepthwiseConvolution2dDescriptor& descriptor,
143 const TensorInfo& weights,
144 const Optional<TensorInfo>& biases,
145 Optional<std::string&> reasonIfUnsupported =
146 EmptyOptional()) const override;
147
arovir011c7c81b2018-10-08 11:34:28 +0100148 bool IsDivisionSupported(const TensorInfo& input0,
149 const TensorInfo& input1,
150 const TensorInfo& output,
151 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
152
josh minor4a3c6102020-01-06 16:40:46 -0600153 bool IsElementwiseUnarySupported(const TensorInfo& input,
154 const TensorInfo& output,
155 const ElementwiseUnaryDescriptor& descriptor,
156 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
157
arovir011c7c81b2018-10-08 11:34:28 +0100158 bool IsFakeQuantizationSupported(const TensorInfo& input,
159 const FakeQuantizationDescriptor& descriptor,
160 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
161
Ryan OSheaf4bfa6a2020-06-10 11:33:37 +0100162 bool IsFillSupported(const TensorInfo& input,
163 const TensorInfo& output,
164 const FillDescriptor& descriptor,
165 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
166
arovir011c7c81b2018-10-08 11:34:28 +0100167 bool IsFloorSupported(const TensorInfo& input,
168 const TensorInfo& output,
169 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
170
171 bool IsFullyConnectedSupported(const TensorInfo& input,
172 const TensorInfo& output,
173 const TensorInfo& weights,
174 const TensorInfo& biases,
175 const FullyConnectedDescriptor& descriptor,
176 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
177
Teresa Charlinb2d3ec52022-04-12 22:07:09 +0100178 bool IsGatherNdSupported(const TensorInfo& input0,
179 const TensorInfo& input1,
180 const TensorInfo& output,
181 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
182
narpra014951d842019-01-18 16:53:53 +0000183 bool IsGatherSupported(const TensorInfo& input0,
184 const TensorInfo& input1,
185 const TensorInfo& output,
Teresa Charlin52664732020-06-29 16:27:03 +0100186 const GatherDescriptor& descriptor,
narpra014951d842019-01-18 16:53:53 +0000187 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
188
arovir011c7c81b2018-10-08 11:34:28 +0100189 bool IsInputSupported(const TensorInfo& input,
190 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
191
Kevin May09ca49c2019-10-09 12:37:34 +0100192 bool IsInstanceNormalizationSupported(const TensorInfo& input,
193 const TensorInfo& output,
194 const InstanceNormalizationDescriptor& descriptor,
195 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
196
arovir011c7c81b2018-10-08 11:34:28 +0100197 bool IsL2NormalizationSupported(const TensorInfo& input,
198 const TensorInfo& output,
199 const L2NormalizationDescriptor& descriptor,
200 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
201
James Conroyaba90cd2020-11-06 16:28:18 +0000202 bool IsLogicalBinarySupported(const TensorInfo& input0,
203 const TensorInfo& input1,
204 const TensorInfo& output,
205 const LogicalBinaryDescriptor& descriptor,
206 Optional<std::string&> reasonIfUnsupported) const override;
207
Aron Virginas-Tare662a942019-10-14 15:12:00 +0100208 bool IsLogSoftmaxSupported(const TensorInfo& input,
209 const TensorInfo& output,
210 const LogSoftmaxDescriptor& descriptor,
211 Optional<std::string&> reasonIfUnsupported) const override;
212
arovir011c7c81b2018-10-08 11:34:28 +0100213 bool IsLstmSupported(const TensorInfo& input,
214 const TensorInfo& outputStateIn,
215 const TensorInfo& cellStateIn,
216 const TensorInfo& scratchBuffer,
217 const TensorInfo& outputStateOut,
218 const TensorInfo& cellStateOut,
219 const TensorInfo& output,
220 const LstmDescriptor& descriptor,
Jan Eilersd01a83c2019-07-03 18:20:40 +0100221 const LstmInputParamsInfo& paramsInfo,
222 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
arovir011c7c81b2018-10-08 11:34:28 +0100223
saoste012df12b32018-11-28 16:57:20 +0000224 bool IsMaximumSupported(const TensorInfo& input0,
225 const TensorInfo& input1,
226 const TensorInfo& output,
227 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
228
arovir011c7c81b2018-10-08 11:34:28 +0100229 bool IsMeanSupported(const TensorInfo& input,
230 const TensorInfo& output,
231 const MeanDescriptor& descriptor,
232 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
233
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000234 bool IsMemCopySupported(const TensorInfo& input,
235 const TensorInfo& output,
236 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
237
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000238 bool IsMinimumSupported(const TensorInfo& input0,
239 const TensorInfo& input1,
240 const TensorInfo& output,
241 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
242
arovir011c7c81b2018-10-08 11:34:28 +0100243 bool IsMultiplicationSupported(const TensorInfo& input0,
244 const TensorInfo& input1,
245 const TensorInfo& output,
246 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
247
248 bool IsNormalizationSupported(const TensorInfo& input,
249 const TensorInfo& output,
250 const NormalizationDescriptor& descriptor,
251 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
252
253 bool IsOutputSupported(const TensorInfo& output,
254 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
255
256 bool IsPadSupported(const TensorInfo& input,
257 const TensorInfo& output,
258 const PadDescriptor& descriptor,
259 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
260
261 bool IsPermuteSupported(const TensorInfo& input,
262 const TensorInfo& output,
263 const PermuteDescriptor& descriptor,
264 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
265
266 bool IsPooling2dSupported(const TensorInfo& input,
267 const TensorInfo& output,
268 const Pooling2dDescriptor& descriptor,
269 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
270
Tamás Nyíri7b885b32021-10-26 14:47:57 +0100271 bool IsPooling3dSupported(const TensorInfo& input,
272 const TensorInfo& output,
273 const Pooling3dDescriptor& descriptor,
274 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
275
Derek Lamberti5f400d62019-03-25 15:41:58 +0000276 bool IsQuantizeSupported(const TensorInfo& input,
277 const TensorInfo& output,
278 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
279
James Conroy4f1f8992020-04-29 20:01:10 +0100280 bool IsQLstmSupported(const TensorInfo& input,
281 const TensorInfo& previousOutputIn,
282 const TensorInfo& previousCellStateIn,
283 const TensorInfo& outputStateOut,
284 const TensorInfo& cellStateOut,
285 const TensorInfo& output,
286 const QLstmDescriptor& descriptor,
287 const LstmInputParamsInfo& paramsInfo,
288 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
289
Finn Williams2605b232020-06-10 15:53:46 +0100290 bool IsRankSupported(const TensorInfo& input,
291 const TensorInfo& output,
292 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
293
Sadik Armagan0c3ea5b2021-02-03 09:29:30 +0000294 bool IsReduceSupported(const TensorInfo& input,
295 const TensorInfo& output,
296 const ReduceDescriptor& descriptor,
297 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
298
arovir011c7c81b2018-10-08 11:34:28 +0100299 bool IsReshapeSupported(const TensorInfo& input,
Kevin Maya023c402019-12-12 17:28:05 +0000300 const TensorInfo& output,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000301 const ReshapeDescriptor& descriptor,
arovir011c7c81b2018-10-08 11:34:28 +0100302 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
303
Teresa Charlin970f43b2019-07-01 13:51:07 +0100304 bool IsResizeSupported(const TensorInfo& input,
305 const TensorInfo& output,
306 const ResizeDescriptor& descriptor,
307 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Keith Davis3ae3f972021-05-21 16:33:48 +0100308
Keith Davis3ae3f972021-05-21 16:33:48 +0100309 bool IsShapeSupported(const TensorInfo& input,
310 const TensorInfo& output,
311 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
312
Aron Virginas-Tar92b9f872019-09-17 17:27:04 +0100313 bool IsSliceSupported(const TensorInfo& input,
314 const TensorInfo& output,
315 const SliceDescriptor& descriptor,
316 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
317
arovir011c7c81b2018-10-08 11:34:28 +0100318 bool IsSoftmaxSupported(const TensorInfo& input,
319 const TensorInfo& output,
320 const SoftmaxDescriptor& descriptor,
321 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
322
Nattapat Chaimanowong3ea76d52018-11-09 14:10:38 +0000323 bool IsSpaceToBatchNdSupported(const TensorInfo& input,
324 const TensorInfo& output,
325 const SpaceToBatchNdDescriptor& descriptor,
326 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
327
Keith Davisa57eccb2019-06-14 17:33:22 +0100328 bool IsSpaceToDepthSupported(const TensorInfo& input,
329 const TensorInfo& output,
330 const SpaceToDepthDescriptor& descriptor,
331 Optional<std::string&> reasonIfUnsupported = EmptyOptional())
332 const override;
333
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100334 bool IsSplitterSupported(const TensorInfo& input,
335 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
336 const ViewsDescriptor& descriptor,
337 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
338
Matthew Jackson81e601c2019-07-11 12:07:09 +0100339 bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
340 const TensorInfo& output,
341 const StackDescriptor& descriptor,
342 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
343
Nattapat Chaimanowong1216b582018-11-23 15:33:41 +0000344 bool IsStridedSliceSupported(const TensorInfo& input,
345 const TensorInfo& output,
346 const StridedSliceDescriptor& descriptor,
347 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
348
arovir011c7c81b2018-10-08 11:34:28 +0100349 bool IsSubtractionSupported(const TensorInfo& input0,
350 const TensorInfo& input1,
351 const TensorInfo& output,
352 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Matteo Martincighab9e5252019-06-13 17:27:46 +0100353
354 bool IsPreluSupported(const TensorInfo& input,
355 const TensorInfo& alpha,
356 const TensorInfo& output,
357 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Aron Virginas-Tar98180ef2019-06-26 15:02:47 +0100358
359 bool IsTransposeConvolution2dSupported(
360 const TensorInfo& input,
361 const TensorInfo& output,
362 const TransposeConvolution2dDescriptor& descriptor,
363 const TensorInfo& weights,
364 const Optional<TensorInfo>& biases,
365 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000366
367 bool IsTransposeSupported(const TensorInfo& input,
368 const TensorInfo& output,
369 const TransposeDescriptor& descriptor,
370 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Narumol Prangnawarate5339e72021-07-28 17:33:28 +0100371
372 bool IsUnidirectionalSequenceLstmSupported(
373 const TensorInfo& input,
374 const TensorInfo& outputStateIn,
375 const TensorInfo& cellStateIn,
Mike Kelly12994962022-04-21 11:57:09 +0100376 const TensorInfo& outputStateOut,
377 const TensorInfo& cellStateOut,
Narumol Prangnawarate5339e72021-07-28 17:33:28 +0100378 const TensorInfo& output,
Narumol Prangnawarate5339e72021-07-28 17:33:28 +0100379 const UnidirectionalSequenceLstmDescriptor& descriptor,
380 const LstmInputParamsInfo& paramsInfo,
381 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
arovir014424b0a2018-10-04 10:46:04 +0100382};
383
Aron Virginas-Tarb5acbb72018-10-15 11:11:51 +0100384} // namespace armnn