Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 526647333571169076f5e72c9fb18c71025bf7c0 / . / src / backends / backendsCommon / LayerSupportBase.hpp
blob: 8d5535ab4e4ddb0893de90ac5027018968234f8a [file] [log] [blame]
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]1//
Teresa Charlin52664732020-06-29 16:27:03 +0100[diff] [blame^]2// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
8#include <armnn/ILayerSupport.hpp>
9
10namespace armnn
11{
12
13class LayerSupportBase : public ILayerSupport
14{
15public:
josh minor4a3c6102020-01-06 16:40:46 -0600[diff] [blame]16 ARMNN_DEPRECATED_MSG("Use IsElementwiseUnarySupported instead")
Kevin May868eb142019-09-04 17:29:31 +0100[diff] [blame]17 bool IsAbsSupported(const TensorInfo& input,
18 const TensorInfo& output,
19 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
20
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]21 bool IsActivationSupported(const TensorInfo& input,
22 const TensorInfo& output,
23 const ActivationDescriptor& descriptor,
24 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
25
26 bool IsAdditionSupported(const TensorInfo& input0,
27 const TensorInfo& input1,
28 const TensorInfo& output,
29 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
30
Nikhil Rajee391d52019-09-05 17:50:44 +0100[diff] [blame]31 bool IsArgMinMaxSupported(const TensorInfo& input,
32 const TensorInfo& output,
33 const ArgMinMaxDescriptor& descriptor,
34 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
35
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]36 bool IsBatchNormalizationSupported(const TensorInfo& input,
37 const TensorInfo& output,
38 const TensorInfo& mean,
39 const TensorInfo& var,
40 const TensorInfo& beta,
41 const TensorInfo& gamma,
42 const BatchNormalizationDescriptor& descriptor,
43 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
44
45 bool IsBatchToSpaceNdSupported(const TensorInfo& input,
46 const TensorInfo& output,
47 const BatchToSpaceNdDescriptor& descriptor,
48 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
49
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100[diff] [blame]50 bool IsComparisonSupported(const TensorInfo& input0,
51 const TensorInfo& input1,
52 const TensorInfo& output,
53 const ComparisonDescriptor& descriptor,
54 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
55
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]56 bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,
57 const TensorInfo& output,
58 const OriginsDescriptor& descriptor,
59 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
60
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]61 bool IsConstantSupported(const TensorInfo& output,
62 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
63
Narumol Prangnawarat7ddbbae2020-03-13 10:26:05 +0000[diff] [blame]64 bool IsConvertBf16ToFp32Supported(const TensorInfo& input,
65 const TensorInfo& output,
66 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
67
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]68 bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
69 const TensorInfo& output,
70 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
71
Narumol Prangnawaratea54a012020-03-16 16:36:10 +0000[diff] [blame]72 bool IsConvertFp32ToBf16Supported(const TensorInfo& input,
73 const TensorInfo& output,
74 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
75
Matteo Martincigh3d6898c2019-01-15 16:11:44 +0000[diff] [blame]76 bool IsConvertFp32ToFp16Supported(
77 const TensorInfo& input,
78 const TensorInfo& output,
79 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]80
81 bool IsConvolution2dSupported(const TensorInfo& input,
82 const TensorInfo& output,
83 const Convolution2dDescriptor& descriptor,
84 const TensorInfo& weights,
85 const Optional<TensorInfo>& biases,
86 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
87
88 bool IsDebugSupported(const TensorInfo& input,
89 const TensorInfo& output,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]90 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
91
Aron Virginas-Tardd6247f2019-09-19 14:31:17 +0100[diff] [blame]92 bool IsDepthToSpaceSupported(const TensorInfo& input,
93 const TensorInfo& output,
94 const DepthToSpaceDescriptor& descriptor,
95 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
96
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]97 bool IsDepthwiseConvolutionSupported(const TensorInfo& input,
98 const TensorInfo& output,
99 const DepthwiseConvolution2dDescriptor& descriptor,
100 const TensorInfo& weights,
101 const Optional<TensorInfo>& biases,
102 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
103
Nattapat Chaimanowonge4294fd2019-03-28 09:56:53 +0000[diff] [blame]104 bool IsDequantizeSupported(const TensorInfo& input,
105 const TensorInfo& output,
106 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
107
Derek Lamberti6a5e5e82019-12-05 14:41:20 +0000[diff] [blame]108 bool IsDetectionPostProcessSupported(const TensorInfo& boxEncodings,
109 const TensorInfo& scores,
110 const TensorInfo& anchors,
111 const TensorInfo& detectionBoxes,
112 const TensorInfo& detectionClasses,
113 const TensorInfo& detectionScores,
114 const TensorInfo& numDetections,
Narumol Prangnawarat94dd5d82019-01-23 18:06:26 +0000[diff] [blame]115 const DetectionPostProcessDescriptor& descriptor,
116 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
117
Matteo Martincigheac246d2019-05-14 09:11:03 +0100[diff] [blame]118 bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo& input,
119 const TensorInfo& output,
120 const DepthwiseConvolution2dDescriptor& descriptor,
121 const TensorInfo& weights,
122 const Optional<TensorInfo>& biases,
123 Optional<std::string&> reasonIfUnsupported =
124 EmptyOptional()) const override;
125
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]126 bool IsDivisionSupported(const TensorInfo& input0,
127 const TensorInfo& input1,
128 const TensorInfo& output,
129 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
130
josh minor4a3c6102020-01-06 16:40:46 -0600[diff] [blame]131 bool IsElementwiseUnarySupported(const TensorInfo& input,
132 const TensorInfo& output,
133 const ElementwiseUnaryDescriptor& descriptor,
134 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
135
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100[diff] [blame]136 ARMNN_DEPRECATED_MSG("Use IsComparisonSupported instead")
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]137 bool IsEqualSupported(const TensorInfo& input0,
138 const TensorInfo& input1,
139 const TensorInfo& output,
140 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
141
142 bool IsFakeQuantizationSupported(const TensorInfo& input,
143 const FakeQuantizationDescriptor& descriptor,
144 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
145
Ryan OSheaec6c6802020-06-05 17:17:06 +0100[diff] [blame]146 virtual bool IsFillSupported(const TensorInfo& input,
147 const TensorInfo& output,
148 const FillDescriptor& descriptor,
149 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
150
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]151 bool IsFloorSupported(const TensorInfo& input,
152 const TensorInfo& output,
153 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
154
155 bool IsFullyConnectedSupported(const TensorInfo& input,
156 const TensorInfo& output,
157 const TensorInfo& weights,
158 const TensorInfo& biases,
159 const FullyConnectedDescriptor& descriptor,
160 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
161
Teresa Charlin52664732020-06-29 16:27:03 +0100[diff] [blame^]162 ARMNN_DEPRECATED_MSG("Use IsGatherSupported with descriptor instead")
narpra01b89b05f2019-01-16 09:53:09 +0000[diff] [blame]163 bool IsGatherSupported(const TensorInfo& input0,
164 const TensorInfo& input1,
165 const TensorInfo& output,
166 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
167
Teresa Charlin52664732020-06-29 16:27:03 +0100[diff] [blame^]168 bool IsGatherSupported(const TensorInfo& input0,
169 const TensorInfo& input1,
170 const TensorInfo& output,
171 const GatherDescriptor& descriptor,
172 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
173
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100[diff] [blame]174 ARMNN_DEPRECATED_MSG("Use IsComparisonSupported instead")
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]175 bool IsGreaterSupported(const TensorInfo& input0,
176 const TensorInfo& input1,
177 const TensorInfo& output,
178 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
179
180 bool IsInputSupported(const TensorInfo& input,
181 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
182
Kevin Mayce5045a2019-10-02 14:07:47 +0100[diff] [blame]183 bool IsInstanceNormalizationSupported(
184 const TensorInfo& input,
185 const TensorInfo& output,
186 const InstanceNormalizationDescriptor& descriptor,
187 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
188
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]189 bool IsL2NormalizationSupported(const TensorInfo& input,
190 const TensorInfo& output,
191 const L2NormalizationDescriptor& descriptor,
192 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
193
Aron Virginas-Tarf982dea2019-10-11 14:07:53 +0100[diff] [blame]194 bool IsLogSoftmaxSupported(const TensorInfo& input,
195 const TensorInfo& output,
196 const LogSoftmaxDescriptor& descriptor,
197 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
198
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]199 bool IsLstmSupported(const TensorInfo& input,
200 const TensorInfo& outputStateIn,
201 const TensorInfo& cellStateIn,
202 const TensorInfo& scratchBuffer,
203 const TensorInfo& outputStateOut,
204 const TensorInfo& cellStateOut,
205 const TensorInfo& output,
206 const LstmDescriptor& descriptor,
Jan Eilersd01a83c2019-07-03 18:20:40 +0100[diff] [blame]207 const LstmInputParamsInfo& paramsInfo,
208 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]209
210 bool IsMaximumSupported(const TensorInfo& input0,
211 const TensorInfo& input1,
212 const TensorInfo& output,
213 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
214
215 bool IsMeanSupported(const TensorInfo& input,
216 const TensorInfo& output,
217 const MeanDescriptor& descriptor,
218 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
219
220 bool IsMemCopySupported(const TensorInfo& input,
221 const TensorInfo& output,
222 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
223
Derek Lambertif674aa02019-08-01 15:56:25 +0100[diff] [blame]224 bool IsMemImportSupported(const TensorInfo& input,
225 const TensorInfo& output,
226 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
227
Nattapat Chaimanowong1f886302019-04-05 13:37:19 +0100[diff] [blame]228 bool IsMergeSupported(const TensorInfo& input0,
229 const TensorInfo& input1,
230 const TensorInfo& output,
231 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
232
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]233 ARMNN_DEPRECATED_MSG("Use IsConcatSupported instead")
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]234 bool IsMergerSupported(const std::vector<const TensorInfo*> inputs,
235 const TensorInfo& output,
236 const OriginsDescriptor& descriptor,
237 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
238
239 bool IsMinimumSupported(const TensorInfo& input0,
240 const TensorInfo& input1,
241 const TensorInfo& output,
242 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
243
244 bool IsMultiplicationSupported(const TensorInfo& input0,
245 const TensorInfo& input1,
246 const TensorInfo& output,
247 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
248
249 bool IsNormalizationSupported(const TensorInfo& input,
250 const TensorInfo& output,
251 const NormalizationDescriptor& descriptor,
252 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
253
254 bool IsOutputSupported(const TensorInfo& output,
255 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
256
257 bool IsPadSupported(const TensorInfo& input,
258 const TensorInfo& output,
259 const PadDescriptor& descriptor,
260 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
261
262 bool IsPermuteSupported(const TensorInfo& input,
263 const TensorInfo& output,
264 const PermuteDescriptor& descriptor,
265 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
266
267 bool IsPooling2dSupported(const TensorInfo& input,
268 const TensorInfo& output,
269 const Pooling2dDescriptor& descriptor,
270 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
271
Matteo Martincigh49124022019-01-11 13:25:59 +0000[diff] [blame]272 bool IsPreCompiledSupported(const TensorInfo& input,
273 const PreCompiledDescriptor& descriptor,
274 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
275
Matteo Martincigh0e406ee2019-06-12 15:42:18 +0100[diff] [blame]276 bool IsPreluSupported(const TensorInfo& input,
277 const TensorInfo& alpha,
278 const TensorInfo& output,
279 Optional<std::string &> reasonIfUnsupported) const override;
280
Derek Lambertia9cca6a2019-03-25 15:41:58 +0000[diff] [blame]281 bool IsQuantizeSupported(const TensorInfo& input,
282 const TensorInfo& output,
283 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
284
James Conroy586a9aa2020-03-20 08:49:33 +0000[diff] [blame]285 bool IsQLstmSupported(const TensorInfo& input,
286 const TensorInfo& previousOutputIn,
287 const TensorInfo& previousCellStateIn,
288 const TensorInfo& outputStateOut,
289 const TensorInfo& cellStateOut,
290 const TensorInfo& output,
291 const QLstmDescriptor& descriptor,
292 const LstmInputParamsInfo& paramsInfo,
293 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
294
James Conroyee18dc82019-07-17 11:27:46 +0100[diff] [blame]295 bool IsQuantizedLstmSupported(const TensorInfo& input,
296 const TensorInfo& previousCellStateIn,
297 const TensorInfo& previousOutputIn,
298 const TensorInfo& cellStateOut,
299 const TensorInfo& output,
300 const QuantizedLstmInputParamsInfo& paramsInfo,
301 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
302
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]303 bool IsReshapeSupported(const TensorInfo& input,
Kevin Maya023c402019-12-12 17:28:05 +0000[diff] [blame]304 const TensorInfo& output,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]305 const ReshapeDescriptor& descriptor,
306 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
307
Teresa Charlina9075df2019-06-27 15:41:57 +0100[diff] [blame]308 bool IsResizeSupported(const TensorInfo& input,
309 const TensorInfo& output,
310 const ResizeDescriptor& descriptor,
311 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
312
Aron Virginas-Tar169d2f12019-07-01 19:01:44 +0100[diff] [blame]313 ARMNN_DEPRECATED_MSG("Use IsResizeSupported instead")
314 bool IsResizeBilinearSupported(const TensorInfo& input,
315 const TensorInfo& output,
316 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
317
josh minor4a3c6102020-01-06 16:40:46 -0600[diff] [blame]318 ARMNN_DEPRECATED_MSG("Use IsElementwiseUnarySupported instead")
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]319 bool IsRsqrtSupported(const TensorInfo& input,
320 const TensorInfo& output,
321 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
322
Aron Virginas-Tar636ab402019-09-16 14:27:45 +0100[diff] [blame]323 bool IsSliceSupported(const TensorInfo& input,
324 const TensorInfo& output,
325 const SliceDescriptor& descriptor,
326 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
327
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]328 bool IsSoftmaxSupported(const TensorInfo& input,
329 const TensorInfo& output,
330 const SoftmaxDescriptor& descriptor,
331 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
332
333 bool IsSpaceToBatchNdSupported(const TensorInfo& input,
334 const TensorInfo& output,
335 const SpaceToBatchNdDescriptor& descriptor,
336 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
337
Aron Virginas-Tar972af152019-06-11 14:14:03 +0100[diff] [blame]338 bool IsSpaceToDepthSupported(const TensorInfo& input,
339 const TensorInfo& output,
340 const SpaceToDepthDescriptor& descriptor,
341 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
342
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100[diff] [blame]343 ARMNN_DEPRECATED_MSG("Use IsSplitterSupported with outputs instead")
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]344 bool IsSplitterSupported(const TensorInfo& input,
345 const ViewsDescriptor& descriptor,
346 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
347
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100[diff] [blame]348 bool IsSplitterSupported(const TensorInfo& input,
349 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
350 const ViewsDescriptor& descriptor,
351 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
352
Matthew Jackson81e601c2019-07-11 12:07:09 +0100[diff] [blame]353 bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
Matthew Jackson2b8c1da2019-07-04 14:59:16 +0100[diff] [blame]354 const TensorInfo& output,
355 const StackDescriptor& descriptor,
356 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
357
Derek Lamberti013c3902019-10-21 10:46:16 +0100[diff] [blame]358 bool IsStandInSupported(const std::vector<const TensorInfo*>& inputs,
359 const std::vector<const TensorInfo*>& outputs,
360 const StandInDescriptor& descriptor,
361 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
362
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]363 bool IsStridedSliceSupported(const TensorInfo& input,
364 const TensorInfo& output,
365 const StridedSliceDescriptor& descriptor,
366 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
367
368 bool IsSubtractionSupported(const TensorInfo& input0,
369 const TensorInfo& input1,
370 const TensorInfo& output,
371 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Sadik Armaganeff363d2019-04-05 15:25:46 +0100[diff] [blame]372
373 bool IsSwitchSupported(const TensorInfo& input0,
374 const TensorInfo& input1,
375 const TensorInfo& output0,
376 const TensorInfo& output1,
377 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Aron Virginas-Tar639fb042019-06-20 14:28:19 +0100[diff] [blame]378
379 bool IsTransposeConvolution2dSupported(
380 const TensorInfo& input,
381 const TensorInfo& output,
382 const TransposeConvolution2dDescriptor& descriptor,
383 const TensorInfo& weights,
384 const Optional<TensorInfo>& biases,
385 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000[diff] [blame]386
387 bool IsTransposeSupported(const TensorInfo& input,
388 const TensorInfo& output,
389 const TransposeDescriptor& descriptor,
390 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
391
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]392};
393
394} // namespace armnn