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