Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 532a29d12d72f54549d8b71edd485c17af65698a / . / src / backends / backendsCommon / LayerSupportBase.hpp
blob: aff45294174f208d50e59c2bc9415012e5e05e77 [file] [log] [blame]
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]1//
2// Copyright © 2017 Arm Ltd. All rights reserved.
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
narpra01b89b05f2019-01-16 09:53:09 +0000[diff] [blame]162 bool IsGatherSupported(const TensorInfo& input0,
163 const TensorInfo& input1,
164 const TensorInfo& output,
165 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
166
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100[diff] [blame]167 ARMNN_DEPRECATED_MSG("Use IsComparisonSupported instead")
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]168 bool IsGreaterSupported(const TensorInfo& input0,
169 const TensorInfo& input1,
170 const TensorInfo& output,
171 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
172
173 bool IsInputSupported(const TensorInfo& input,
174 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
175
Kevin Mayce5045a2019-10-02 14:07:47 +0100[diff] [blame]176 bool IsInstanceNormalizationSupported(
177 const TensorInfo& input,
178 const TensorInfo& output,
179 const InstanceNormalizationDescriptor& descriptor,
180 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
181
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]182 bool IsL2NormalizationSupported(const TensorInfo& input,
183 const TensorInfo& output,
184 const L2NormalizationDescriptor& descriptor,
185 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
186
Aron Virginas-Tarf982dea2019-10-11 14:07:53 +0100[diff] [blame]187 bool IsLogSoftmaxSupported(const TensorInfo& input,
188 const TensorInfo& output,
189 const LogSoftmaxDescriptor& descriptor,
190 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
191
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]192 bool IsLstmSupported(const TensorInfo& input,
193 const TensorInfo& outputStateIn,
194 const TensorInfo& cellStateIn,
195 const TensorInfo& scratchBuffer,
196 const TensorInfo& outputStateOut,
197 const TensorInfo& cellStateOut,
198 const TensorInfo& output,
199 const LstmDescriptor& descriptor,
Jan Eilersd01a83c2019-07-03 18:20:40 +0100[diff] [blame]200 const LstmInputParamsInfo& paramsInfo,
201 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]202
203 bool IsMaximumSupported(const TensorInfo& input0,
204 const TensorInfo& input1,
205 const TensorInfo& output,
206 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
207
208 bool IsMeanSupported(const TensorInfo& input,
209 const TensorInfo& output,
210 const MeanDescriptor& descriptor,
211 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
212
213 bool IsMemCopySupported(const TensorInfo& input,
214 const TensorInfo& output,
215 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
216
Derek Lambertif674aa02019-08-01 15:56:25 +0100[diff] [blame]217 bool IsMemImportSupported(const TensorInfo& input,
218 const TensorInfo& output,
219 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
220
Nattapat Chaimanowong1f886302019-04-05 13:37:19 +0100[diff] [blame]221 bool IsMergeSupported(const TensorInfo& input0,
222 const TensorInfo& input1,
223 const TensorInfo& output,
224 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
225
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]226 ARMNN_DEPRECATED_MSG("Use IsConcatSupported instead")
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]227 bool IsMergerSupported(const std::vector<const TensorInfo*> inputs,
228 const TensorInfo& output,
229 const OriginsDescriptor& descriptor,
230 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
231
232 bool IsMinimumSupported(const TensorInfo& input0,
233 const TensorInfo& input1,
234 const TensorInfo& output,
235 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
236
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
Matteo Martincigh49124022019-01-11 13:25:59 +0000[diff] [blame]265 bool IsPreCompiledSupported(const TensorInfo& input,
266 const PreCompiledDescriptor& descriptor,
267 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
268
Matteo Martincigh0e406ee2019-06-12 15:42:18 +0100[diff] [blame]269 bool IsPreluSupported(const TensorInfo& input,
270 const TensorInfo& alpha,
271 const TensorInfo& output,
272 Optional<std::string &> reasonIfUnsupported) const override;
273
Derek Lambertia9cca6a2019-03-25 15:41:58 +0000[diff] [blame]274 bool IsQuantizeSupported(const TensorInfo& input,
275 const TensorInfo& output,
276 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
277
James Conroy586a9aa2020-03-20 08:49:33 +0000[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,
286 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
287
James Conroyee18dc82019-07-17 11:27:46 +0100[diff] [blame]288 bool IsQuantizedLstmSupported(const TensorInfo& input,
289 const TensorInfo& previousCellStateIn,
290 const TensorInfo& previousOutputIn,
291 const TensorInfo& cellStateOut,
292 const TensorInfo& output,
293 const QuantizedLstmInputParamsInfo& paramsInfo,
294 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
295
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]296 bool IsReshapeSupported(const TensorInfo& input,
Kevin Maya023c402019-12-12 17:28:05 +0000[diff] [blame]297 const TensorInfo& output,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]298 const ReshapeDescriptor& descriptor,
299 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
300
Teresa Charlina9075df2019-06-27 15:41:57 +0100[diff] [blame]301 bool IsResizeSupported(const TensorInfo& input,
302 const TensorInfo& output,
303 const ResizeDescriptor& descriptor,
304 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
305
Aron Virginas-Tar169d2f12019-07-01 19:01:44 +0100[diff] [blame]306 ARMNN_DEPRECATED_MSG("Use IsResizeSupported instead")
307 bool IsResizeBilinearSupported(const TensorInfo& input,
308 const TensorInfo& output,
309 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
310
josh minor4a3c6102020-01-06 16:40:46 -0600[diff] [blame]311 ARMNN_DEPRECATED_MSG("Use IsElementwiseUnarySupported instead")
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]312 bool IsRsqrtSupported(const TensorInfo& input,
313 const TensorInfo& output,
314 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
315
Aron Virginas-Tar636ab402019-09-16 14:27:45 +0100[diff] [blame]316 bool IsSliceSupported(const TensorInfo& input,
317 const TensorInfo& output,
318 const SliceDescriptor& descriptor,
319 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
320
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]321 bool IsSoftmaxSupported(const TensorInfo& input,
322 const TensorInfo& output,
323 const SoftmaxDescriptor& descriptor,
324 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
325
326 bool IsSpaceToBatchNdSupported(const TensorInfo& input,
327 const TensorInfo& output,
328 const SpaceToBatchNdDescriptor& descriptor,
329 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
330
Aron Virginas-Tar972af152019-06-11 14:14:03 +0100[diff] [blame]331 bool IsSpaceToDepthSupported(const TensorInfo& input,
332 const TensorInfo& output,
333 const SpaceToDepthDescriptor& descriptor,
334 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
335
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100[diff] [blame]336 ARMNN_DEPRECATED_MSG("Use IsSplitterSupported with outputs instead")
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]337 bool IsSplitterSupported(const TensorInfo& input,
338 const ViewsDescriptor& descriptor,
339 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
340
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100[diff] [blame]341 bool IsSplitterSupported(const TensorInfo& input,
342 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
343 const ViewsDescriptor& descriptor,
344 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
345
Matthew Jackson81e601c2019-07-11 12:07:09 +0100[diff] [blame]346 bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
Matthew Jackson2b8c1da2019-07-04 14:59:16 +0100[diff] [blame]347 const TensorInfo& output,
348 const StackDescriptor& descriptor,
349 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
350
Derek Lamberti013c3902019-10-21 10:46:16 +0100[diff] [blame]351 bool IsStandInSupported(const std::vector<const TensorInfo*>& inputs,
352 const std::vector<const TensorInfo*>& outputs,
353 const StandInDescriptor& descriptor,
354 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
355
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]356 bool IsStridedSliceSupported(const TensorInfo& input,
357 const TensorInfo& output,
358 const StridedSliceDescriptor& descriptor,
359 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
360
361 bool IsSubtractionSupported(const TensorInfo& input0,
362 const TensorInfo& input1,
363 const TensorInfo& output,
364 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Sadik Armaganeff363d2019-04-05 15:25:46 +0100[diff] [blame]365
366 bool IsSwitchSupported(const TensorInfo& input0,
367 const TensorInfo& input1,
368 const TensorInfo& output0,
369 const TensorInfo& output1,
370 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Aron Virginas-Tar639fb042019-06-20 14:28:19 +0100[diff] [blame]371
372 bool IsTransposeConvolution2dSupported(
373 const TensorInfo& input,
374 const TensorInfo& output,
375 const TransposeConvolution2dDescriptor& descriptor,
376 const TensorInfo& weights,
377 const Optional<TensorInfo>& biases,
378 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000[diff] [blame]379
380 bool IsTransposeSupported(const TensorInfo& input,
381 const TensorInfo& output,
382 const TransposeDescriptor& descriptor,
383 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
384
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]385};
386
387} // namespace armnn