blob: 7a95bb06ac2c8653318c5fb2a689f7ffc17b99c6 [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:
josh minor4a3c6102020-01-06 16:40:46 -060015 ARMNN_DEPRECATED_MSG("Use IsElementwiseUnarySupported instead")
Sadik Armagan9199e582019-09-05 17:35:31 +010016 bool IsAbsSupported(const TensorInfo& input,
17 const TensorInfo& output,
18 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
19
arovir011c7c81b2018-10-08 11:34:28 +010020 bool IsActivationSupported(const TensorInfo& input,
21 const TensorInfo& output,
22 const ActivationDescriptor& descriptor,
23 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
24
25 bool IsAdditionSupported(const TensorInfo& input0,
26 const TensorInfo& input1,
27 const TensorInfo& output,
28 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
29
Nikhil Raj68c2c902019-09-19 11:21:11 +010030 bool IsArgMinMaxSupported(const TensorInfo& input,
31 const TensorInfo& output,
32 const ArgMinMaxDescriptor& descriptor,
33 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
34
arovir011c7c81b2018-10-08 11:34:28 +010035 bool IsBatchNormalizationSupported(const TensorInfo& input,
36 const TensorInfo& output,
37 const TensorInfo& mean,
38 const TensorInfo& var,
39 const TensorInfo& beta,
40 const TensorInfo& gamma,
41 const BatchNormalizationDescriptor& descriptor,
42 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
43
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +000044 bool IsBatchToSpaceNdSupported(const TensorInfo& input,
45 const TensorInfo& output,
46 const BatchToSpaceNdDescriptor& descriptor,
47 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
48
mathad01b392e982021-04-07 12:07:30 +010049 bool IsCastSupported(const TensorInfo& input,
50 const TensorInfo& output,
51 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
52
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +010053 bool IsComparisonSupported(const TensorInfo& input0,
54 const TensorInfo& input1,
55 const TensorInfo& output,
56 const ComparisonDescriptor& descriptor,
57 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
58
Jim Flynn906f9462019-05-10 13:55:21 +010059 bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,
60 const TensorInfo& output,
Jim Flynne242f2d2019-05-22 14:24:13 +010061 const ConcatDescriptor& descriptor,
Jim Flynn906f9462019-05-10 13:55:21 +010062 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
63
arovir011c7c81b2018-10-08 11:34:28 +010064 bool IsConstantSupported(const TensorInfo& output,
65 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
66
Narumol Prangnawarat7ddbbae2020-03-13 10:26:05 +000067 bool IsConvertBf16ToFp32Supported(const TensorInfo& input,
68 const TensorInfo& output,
69 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
70
arovir011c7c81b2018-10-08 11:34:28 +010071 bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
72 const TensorInfo& output,
73 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
74
Narumol Prangnawaratea54a012020-03-16 16:36:10 +000075 bool IsConvertFp32ToBf16Supported(const TensorInfo& input,
76 const TensorInfo& output,
77 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
78
arovir011c7c81b2018-10-08 11:34:28 +010079 bool IsConvertFp32ToFp16Supported(const TensorInfo& input,
80 const TensorInfo& output,
81 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
82
83 bool IsConvolution2dSupported(const TensorInfo& input,
84 const TensorInfo& output,
85 const Convolution2dDescriptor& descriptor,
86 const TensorInfo& weights,
87 const Optional<TensorInfo>& biases,
88 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
89
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +000090 bool IsDebugSupported(const TensorInfo& input,
91 const TensorInfo& output,
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +000092 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
93
Aron Virginas-Tar73f66422019-09-23 19:11:59 +010094 bool IsDepthToSpaceSupported(const TensorInfo& input,
95 const TensorInfo& output,
96 const DepthToSpaceDescriptor& descriptor,
97 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
98
arovir011c7c81b2018-10-08 11:34:28 +010099 bool IsDepthwiseConvolutionSupported(const TensorInfo& input,
100 const TensorInfo& output,
101 const DepthwiseConvolution2dDescriptor& descriptor,
102 const TensorInfo& weights,
103 const Optional<TensorInfo>& biases,
104 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
105
Nattapat Chaimanowong8a54ac02019-03-29 15:25:04 +0000106 bool IsDequantizeSupported(const TensorInfo& input,
107 const TensorInfo& output,
108 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
109
Derek Lamberti6a5e5e82019-12-05 14:41:20 +0000110 bool IsDetectionPostProcessSupported(const TensorInfo& boxEncodings,
111 const TensorInfo& scores,
112 const TensorInfo& anchors,
113 const TensorInfo& detectionBoxes,
114 const TensorInfo& detectionClasses,
115 const TensorInfo& detectionScores,
116 const TensorInfo& numDetections,
Narumol Prangnawaratbc67cef2019-01-31 15:31:54 +0000117 const DetectionPostProcessDescriptor& descriptor,
118 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
119
Pablo Tellof0bd6832019-04-26 17:58:13 +0100120 bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo& input,
121 const TensorInfo& output,
122 const DepthwiseConvolution2dDescriptor& descriptor,
123 const TensorInfo& weights,
124 const Optional<TensorInfo>& biases,
125 Optional<std::string&> reasonIfUnsupported =
126 EmptyOptional()) const override;
127
arovir011c7c81b2018-10-08 11:34:28 +0100128 bool IsDivisionSupported(const TensorInfo& input0,
129 const TensorInfo& input1,
130 const TensorInfo& output,
131 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
132
josh minor4a3c6102020-01-06 16:40:46 -0600133 bool IsElementwiseUnarySupported(const TensorInfo& input,
134 const TensorInfo& output,
135 const ElementwiseUnaryDescriptor& descriptor,
136 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
137
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100138 ARMNN_DEPRECATED_MSG("Use IsComparisonSupported instead")
FrancisMurtagh30cdfca2018-12-18 12:57:35 +0000139 bool IsEqualSupported(const TensorInfo& input0,
140 const TensorInfo& input1,
141 const TensorInfo& output,
142 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
143
arovir011c7c81b2018-10-08 11:34:28 +0100144 bool IsFakeQuantizationSupported(const TensorInfo& input,
145 const FakeQuantizationDescriptor& descriptor,
146 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
147
Ryan OSheaf4bfa6a2020-06-10 11:33:37 +0100148 bool IsFillSupported(const TensorInfo& input,
149 const TensorInfo& output,
150 const FillDescriptor& descriptor,
151 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
152
arovir011c7c81b2018-10-08 11:34:28 +0100153 bool IsFloorSupported(const TensorInfo& input,
154 const TensorInfo& output,
155 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
156
157 bool IsFullyConnectedSupported(const TensorInfo& input,
158 const TensorInfo& output,
159 const TensorInfo& weights,
160 const TensorInfo& biases,
161 const FullyConnectedDescriptor& descriptor,
162 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
163
narpra014951d842019-01-18 16:53:53 +0000164 bool IsGatherSupported(const TensorInfo& input0,
165 const TensorInfo& input1,
166 const TensorInfo& output,
Teresa Charlin52664732020-06-29 16:27:03 +0100167 const GatherDescriptor& descriptor,
narpra014951d842019-01-18 16:53:53 +0000168 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
169
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100170 ARMNN_DEPRECATED_MSG("Use IsComparisonSupported instead")
FrancisMurtagh878f0232018-12-19 10:56:15 +0000171 bool IsGreaterSupported(const TensorInfo& input0,
172 const TensorInfo& input1,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000173 const TensorInfo& output,
FrancisMurtagh878f0232018-12-19 10:56:15 +0000174 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
175
arovir011c7c81b2018-10-08 11:34:28 +0100176 bool IsInputSupported(const TensorInfo& input,
177 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
178
Kevin May09ca49c2019-10-09 12:37:34 +0100179 bool IsInstanceNormalizationSupported(const TensorInfo& input,
180 const TensorInfo& output,
181 const InstanceNormalizationDescriptor& descriptor,
182 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
183
arovir011c7c81b2018-10-08 11:34:28 +0100184 bool IsL2NormalizationSupported(const TensorInfo& input,
185 const TensorInfo& output,
186 const L2NormalizationDescriptor& descriptor,
187 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
188
James Conroyaba90cd2020-11-06 16:28:18 +0000189 bool IsLogicalBinarySupported(const TensorInfo& input0,
190 const TensorInfo& input1,
191 const TensorInfo& output,
192 const LogicalBinaryDescriptor& descriptor,
193 Optional<std::string&> reasonIfUnsupported) const override;
194
Aron Virginas-Tare662a942019-10-14 15:12:00 +0100195 bool IsLogSoftmaxSupported(const TensorInfo& input,
196 const TensorInfo& output,
197 const LogSoftmaxDescriptor& descriptor,
198 Optional<std::string&> reasonIfUnsupported) const override;
199
arovir011c7c81b2018-10-08 11:34:28 +0100200 bool IsLstmSupported(const TensorInfo& input,
201 const TensorInfo& outputStateIn,
202 const TensorInfo& cellStateIn,
203 const TensorInfo& scratchBuffer,
204 const TensorInfo& outputStateOut,
205 const TensorInfo& cellStateOut,
206 const TensorInfo& output,
207 const LstmDescriptor& descriptor,
Jan Eilersd01a83c2019-07-03 18:20:40 +0100208 const LstmInputParamsInfo& paramsInfo,
209 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
arovir011c7c81b2018-10-08 11:34:28 +0100210
saoste012df12b32018-11-28 16:57:20 +0000211 bool IsMaximumSupported(const TensorInfo& input0,
212 const TensorInfo& input1,
213 const TensorInfo& output,
214 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
215
arovir011c7c81b2018-10-08 11:34:28 +0100216 bool IsMeanSupported(const TensorInfo& input,
217 const TensorInfo& output,
218 const MeanDescriptor& descriptor,
219 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
220
Jim Flynn906f9462019-05-10 13:55:21 +0100221 ARMNN_DEPRECATED_MSG("Use IsConcatSupported instead")
arovir011c7c81b2018-10-08 11:34:28 +0100222 bool IsMergerSupported(const std::vector<const TensorInfo*> inputs,
Nikhil Raj8599a412018-11-19 14:51:07 +0000223 const TensorInfo& output,
Jim Flynne242f2d2019-05-22 14:24:13 +0100224 const MergerDescriptor& descriptor,
arovir011c7c81b2018-10-08 11:34:28 +0100225 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
226
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000227 bool IsMemCopySupported(const TensorInfo& input,
228 const TensorInfo& output,
229 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
230
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000231 bool IsMinimumSupported(const TensorInfo& input0,
232 const TensorInfo& input1,
233 const TensorInfo& output,
234 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
235
arovir011c7c81b2018-10-08 11:34:28 +0100236 bool IsMultiplicationSupported(const TensorInfo& input0,
237 const TensorInfo& input1,
238 const TensorInfo& output,
239 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
240
241 bool IsNormalizationSupported(const TensorInfo& input,
242 const TensorInfo& output,
243 const NormalizationDescriptor& descriptor,
244 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
245
246 bool IsOutputSupported(const TensorInfo& output,
247 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
248
249 bool IsPadSupported(const TensorInfo& input,
250 const TensorInfo& output,
251 const PadDescriptor& descriptor,
252 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
253
254 bool IsPermuteSupported(const TensorInfo& input,
255 const TensorInfo& output,
256 const PermuteDescriptor& descriptor,
257 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
258
259 bool IsPooling2dSupported(const TensorInfo& input,
260 const TensorInfo& output,
261 const Pooling2dDescriptor& descriptor,
262 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
263
Derek Lamberti5f400d62019-03-25 15:41:58 +0000264 bool IsQuantizeSupported(const TensorInfo& input,
265 const TensorInfo& output,
266 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
267
James Conroy4f1f8992020-04-29 20:01:10 +0100268 bool IsQLstmSupported(const TensorInfo& input,
269 const TensorInfo& previousOutputIn,
270 const TensorInfo& previousCellStateIn,
271 const TensorInfo& outputStateOut,
272 const TensorInfo& cellStateOut,
273 const TensorInfo& output,
274 const QLstmDescriptor& descriptor,
275 const LstmInputParamsInfo& paramsInfo,
276 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
277
Finn Williams2605b232020-06-10 15:53:46 +0100278 bool IsRankSupported(const TensorInfo& input,
279 const TensorInfo& output,
280 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
281
Sadik Armagan0c3ea5b2021-02-03 09:29:30 +0000282 bool IsReduceSupported(const TensorInfo& input,
283 const TensorInfo& output,
284 const ReduceDescriptor& descriptor,
285 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
286
arovir011c7c81b2018-10-08 11:34:28 +0100287 bool IsReshapeSupported(const TensorInfo& input,
Kevin Maya023c402019-12-12 17:28:05 +0000288 const TensorInfo& output,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000289 const ReshapeDescriptor& descriptor,
arovir011c7c81b2018-10-08 11:34:28 +0100290 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
291
292 bool IsResizeBilinearSupported(const TensorInfo& input,
Sadik Armaganc625f002018-12-17 11:32:16 +0000293 const TensorInfo& output,
arovir011c7c81b2018-10-08 11:34:28 +0100294 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
295
Teresa Charlin970f43b2019-07-01 13:51:07 +0100296 bool IsResizeSupported(const TensorInfo& input,
297 const TensorInfo& output,
298 const ResizeDescriptor& descriptor,
299 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
josh minor4a3c6102020-01-06 16:40:46 -0600300
301 ARMNN_DEPRECATED_MSG("Use IsElementwiseUnarySupported instead")
Mohamed Nour Abouelseouda1d3c6a2018-12-27 12:39:16 +0000302 bool IsRsqrtSupported(const TensorInfo& input,
303 const TensorInfo& output,
304 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
305
Aron Virginas-Tar92b9f872019-09-17 17:27:04 +0100306 bool IsSliceSupported(const TensorInfo& input,
307 const TensorInfo& output,
308 const SliceDescriptor& descriptor,
309 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
310
arovir011c7c81b2018-10-08 11:34:28 +0100311 bool IsSoftmaxSupported(const TensorInfo& input,
312 const TensorInfo& output,
313 const SoftmaxDescriptor& descriptor,
314 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
315
Nattapat Chaimanowong3ea76d52018-11-09 14:10:38 +0000316 bool IsSpaceToBatchNdSupported(const TensorInfo& input,
317 const TensorInfo& output,
318 const SpaceToBatchNdDescriptor& descriptor,
319 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
320
Keith Davisa57eccb2019-06-14 17:33:22 +0100321 bool IsSpaceToDepthSupported(const TensorInfo& input,
322 const TensorInfo& output,
323 const SpaceToDepthDescriptor& descriptor,
324 Optional<std::string&> reasonIfUnsupported = EmptyOptional())
325 const override;
326
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100327 ARMNN_DEPRECATED_MSG("Use IsSplitterSupported with outputs instead")
arovir011c7c81b2018-10-08 11:34:28 +0100328 bool IsSplitterSupported(const TensorInfo& input,
329 const ViewsDescriptor& descriptor,
330 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
331
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100332 bool IsSplitterSupported(const TensorInfo& input,
333 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
334 const ViewsDescriptor& descriptor,
335 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
336
Matthew Jackson81e601c2019-07-11 12:07:09 +0100337 bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
338 const TensorInfo& output,
339 const StackDescriptor& descriptor,
340 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
341
Nattapat Chaimanowong1216b582018-11-23 15:33:41 +0000342 bool IsStridedSliceSupported(const TensorInfo& input,
343 const TensorInfo& output,
344 const StridedSliceDescriptor& descriptor,
345 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
346
arovir011c7c81b2018-10-08 11:34:28 +0100347 bool IsSubtractionSupported(const TensorInfo& input0,
348 const TensorInfo& input1,
349 const TensorInfo& output,
350 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Matteo Martincighab9e5252019-06-13 17:27:46 +0100351
352 bool IsPreluSupported(const TensorInfo& input,
353 const TensorInfo& alpha,
354 const TensorInfo& output,
355 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Aron Virginas-Tar98180ef2019-06-26 15:02:47 +0100356
357 bool IsTransposeConvolution2dSupported(
358 const TensorInfo& input,
359 const TensorInfo& output,
360 const TransposeConvolution2dDescriptor& descriptor,
361 const TensorInfo& weights,
362 const Optional<TensorInfo>& biases,
363 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000364
365 bool IsTransposeSupported(const TensorInfo& input,
366 const TensorInfo& output,
367 const TransposeDescriptor& descriptor,
368 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
arovir014424b0a2018-10-04 10:46:04 +0100369};
370
Aron Virginas-Tarb5acbb72018-10-15 11:11:51 +0100371} // namespace armnn