ArmNN
 22.08
LayerSupportBase.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
9 
10 namespace armnn
11 {
12 
14 {
15 public:
16 
17  bool IsLayerSupported(const LayerType& type,
18  const std::vector<TensorInfo>& infos,
20  const Optional<LstmInputParamsInfo>& lstmParamsInfo = EmptyOptional(),
21  const Optional<QuantizedLstmInputParamsInfo>& quantizedLstmParamsInfo = EmptyOptional(),
23 
24  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
25  bool IsActivationSupported(const TensorInfo& input,
26  const TensorInfo& output,
27  const ActivationDescriptor& descriptor,
29 
30  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
31  bool IsAdditionSupported(const TensorInfo& input0,
32  const TensorInfo& input1,
33  const TensorInfo& output,
35 
36  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
37  bool IsArgMinMaxSupported(const TensorInfo& input,
38  const TensorInfo& output,
39  const ArgMinMaxDescriptor& descriptor,
41 
42  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
43  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,
51 
52  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
53  bool IsBatchToSpaceNdSupported(const TensorInfo& input,
54  const TensorInfo& output,
55  const BatchToSpaceNdDescriptor& descriptor,
57 
58  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
59  bool IsCastSupported(const TensorInfo& input,
60  const TensorInfo& output,
62 
63  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
64  bool IsChannelShuffleSupported(const TensorInfo& input,
65  const TensorInfo& output,
66  const ChannelShuffleDescriptor& descriptor,
68 
69  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
70  bool IsComparisonSupported(const TensorInfo& input0,
71  const TensorInfo& input1,
72  const TensorInfo& output,
73  const ComparisonDescriptor& descriptor,
75 
76  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
77  bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,
78  const TensorInfo& output,
79  const OriginsDescriptor& descriptor,
81 
82  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
83  bool IsConstantSupported(const TensorInfo& output,
85 
86  bool IsConvertBf16ToFp32Supported(const TensorInfo& input,
87  const TensorInfo& output,
89 
90  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
91  bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
92  const TensorInfo& output,
94 
95  bool IsConvertFp32ToBf16Supported(const TensorInfo& input,
96  const TensorInfo& output,
98 
99  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
101  const TensorInfo& input,
102  const TensorInfo& output,
104 
105  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
106  bool IsConvolution2dSupported(const TensorInfo& input,
107  const TensorInfo& output,
108  const Convolution2dDescriptor& descriptor,
109  const TensorInfo& weights,
112 
113  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
114  bool IsConvolution3dSupported(const TensorInfo& input,
115  const TensorInfo& output,
116  const Convolution3dDescriptor& descriptor,
117  const TensorInfo& weights,
118  const Optional<TensorInfo>& biases,
120 
121  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
122  bool IsDebugSupported(const TensorInfo& input,
123  const TensorInfo& output,
125 
126  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
127  bool IsDepthToSpaceSupported(const TensorInfo& input,
128  const TensorInfo& output,
129  const DepthToSpaceDescriptor& descriptor,
131 
132  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
134  const TensorInfo& output,
135  const DepthwiseConvolution2dDescriptor& descriptor,
136  const TensorInfo& weights,
137  const Optional<TensorInfo>& biases,
139 
140  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
141  bool IsDequantizeSupported(const TensorInfo& input,
142  const TensorInfo& output,
144 
145  bool IsDetectionPostProcessSupported(const TensorInfo& boxEncodings,
146  const TensorInfo& scores,
147  const TensorInfo& anchors,
148  const TensorInfo& detectionBoxes,
151  const TensorInfo& numDetections,
152  const DetectionPostProcessDescriptor& descriptor,
154 
155  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
157  const TensorInfo& output,
158  const DepthwiseConvolution2dDescriptor& descriptor,
159  const TensorInfo& weights,
160  const Optional<TensorInfo>& biases,
162  EmptyOptional()) const override;
163 
164  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
165  bool IsDivisionSupported(const TensorInfo& input0,
166  const TensorInfo& input1,
167  const TensorInfo& output,
169 
170  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
171  bool IsElementwiseUnarySupported(const TensorInfo& input,
172  const TensorInfo& output,
173  const ElementwiseUnaryDescriptor& descriptor,
175 
176  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
177  bool IsFakeQuantizationSupported(const TensorInfo& input,
178  const FakeQuantizationDescriptor& descriptor,
180 
181  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
182  virtual bool IsFillSupported(const TensorInfo& input,
183  const TensorInfo& output,
184  const FillDescriptor& descriptor,
186 
187  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
188  bool IsFloorSupported(const TensorInfo& input,
189  const TensorInfo& output,
191 
192  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
193  bool IsFullyConnectedSupported(const TensorInfo& input,
194  const TensorInfo& output,
195  const TensorInfo& weights,
196  const TensorInfo& biases,
197  const FullyConnectedDescriptor& descriptor,
199 
200  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
201  bool IsGatherSupported(const TensorInfo& input0,
202  const TensorInfo& input1,
203  const TensorInfo& output,
204  const GatherDescriptor& descriptor,
206 
207  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
208  bool IsInputSupported(const TensorInfo& input,
210 
211  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
213  const TensorInfo& input,
214  const TensorInfo& output,
215  const InstanceNormalizationDescriptor& descriptor,
217 
218  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
219  bool IsL2NormalizationSupported(const TensorInfo& input,
220  const TensorInfo& output,
221  const L2NormalizationDescriptor& descriptor,
223 
224  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
225  bool IsLogicalBinarySupported(const TensorInfo& input0,
226  const TensorInfo& input1,
227  const TensorInfo& output,
228  const LogicalBinaryDescriptor& descriptor,
230 
231  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
232  bool IsLogicalUnarySupported(const TensorInfo& input,
233  const TensorInfo& output,
234  const ElementwiseUnaryDescriptor& descriptor,
236 
237  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
238  bool IsLogSoftmaxSupported(const TensorInfo& input,
239  const TensorInfo& output,
240  const LogSoftmaxDescriptor& descriptor,
242 
243  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
244  bool IsLstmSupported(const TensorInfo& input,
245  const TensorInfo& outputStateIn,
246  const TensorInfo& cellStateIn,
247  const TensorInfo& scratchBuffer,
248  const TensorInfo& outputStateOut,
249  const TensorInfo& cellStateOut,
250  const TensorInfo& output,
251  const LstmDescriptor& descriptor,
254 
255  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
256  bool IsMaximumSupported(const TensorInfo& input0,
257  const TensorInfo& input1,
258  const TensorInfo& output,
260 
261  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
262  bool IsMeanSupported(const TensorInfo& input,
263  const TensorInfo& output,
264  const MeanDescriptor& descriptor,
266 
267  bool IsMemCopySupported(const TensorInfo& input,
268  const TensorInfo& output,
270 
271  bool IsMemImportSupported(const TensorInfo& input,
272  const TensorInfo& output,
274 
275  bool IsMergeSupported(const TensorInfo& input0,
276  const TensorInfo& input1,
277  const TensorInfo& output,
279 
280  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
281  bool IsMinimumSupported(const TensorInfo& input0,
282  const TensorInfo& input1,
283  const TensorInfo& output,
285 
286  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
287  bool IsMultiplicationSupported(const TensorInfo& input0,
288  const TensorInfo& input1,
289  const TensorInfo& output,
291 
292  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
293  bool IsNormalizationSupported(const TensorInfo& input,
294  const TensorInfo& output,
295  const NormalizationDescriptor& descriptor,
297 
298  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
299  bool IsOutputSupported(const TensorInfo& output,
301 
302  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
303  bool IsPadSupported(const TensorInfo& input,
304  const TensorInfo& output,
305  const PadDescriptor& descriptor,
307 
308  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
309  bool IsPermuteSupported(const TensorInfo& input,
310  const TensorInfo& output,
311  const PermuteDescriptor& descriptor,
313 
314  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
315  bool IsPooling2dSupported(const TensorInfo& input,
316  const TensorInfo& output,
317  const Pooling2dDescriptor& descriptor,
319 
320  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
321  bool IsPooling3dSupported(const TensorInfo& input,
322  const TensorInfo& output,
323  const Pooling3dDescriptor& descriptor,
325 
326  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
327  bool IsPreCompiledSupported(const TensorInfo& input,
328  const PreCompiledDescriptor& descriptor,
330 
331  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
332  bool IsPreluSupported(const TensorInfo& input,
333  const TensorInfo& alpha,
334  const TensorInfo& output,
336 
337  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
338  bool IsQuantizeSupported(const TensorInfo& input,
339  const TensorInfo& output,
340  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
341 
342  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
343  bool IsQLstmSupported(const TensorInfo& input,
346  const TensorInfo& outputStateOut,
347  const TensorInfo& cellStateOut,
348  const TensorInfo& output,
349  const QLstmDescriptor& descriptor,
350  const LstmInputParamsInfo& paramsInfo,
351  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
352 
353  bool IsQuantizedLstmSupported(const TensorInfo& input,
354  const TensorInfo& previousCellStateIn,
355  const TensorInfo& previousOutputIn,
356  const TensorInfo& cellStateOut,
357  const TensorInfo& output,
358  const QuantizedLstmInputParamsInfo& paramsInfo,
359  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
360 
361  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
362  bool IsRankSupported(const TensorInfo& input,
363  const TensorInfo& output,
364  Optional<std::string&> reasonIfUnsupported) const override;
365 
366  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
367  bool IsReduceSupported(const TensorInfo& input,
368  const TensorInfo& output,
369  const ReduceDescriptor& descriptor,
370  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
371 
372  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
373  bool IsReshapeSupported(const TensorInfo& input,
374  const TensorInfo& output,
375  const ReshapeDescriptor& descriptor,
376  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
377 
378  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
379  bool IsResizeSupported(const TensorInfo& input,
380  const TensorInfo& output,
381  const ResizeDescriptor& descriptor,
382  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
383 
384  bool IsShapeSupported(const TensorInfo& input,
385  const TensorInfo& output,
386  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
387 
388  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
389  bool IsSliceSupported(const TensorInfo& input,
390  const TensorInfo& output,
391  const SliceDescriptor& descriptor,
392  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
393 
394  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
395  bool IsSoftmaxSupported(const TensorInfo& input,
396  const TensorInfo& output,
397  const SoftmaxDescriptor& descriptor,
398  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
399 
400  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
401  bool IsSpaceToBatchNdSupported(const TensorInfo& input,
402  const TensorInfo& output,
403  const SpaceToBatchNdDescriptor& descriptor,
404  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
405 
406  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
407  bool IsSpaceToDepthSupported(const TensorInfo& input,
408  const TensorInfo& output,
409  const SpaceToDepthDescriptor& descriptor,
410  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
411 
412  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
413  bool IsSplitterSupported(const TensorInfo& input,
414  const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
415  const ViewsDescriptor& descriptor,
416  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
417 
418  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
419  bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
420  const TensorInfo& output,
421  const StackDescriptor& descriptor,
422  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
423 
424  bool IsStandInSupported(const std::vector<const TensorInfo*>& inputs,
425  const std::vector<const TensorInfo*>& outputs,
426  const StandInDescriptor& descriptor,
427  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
428 
429  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
430  bool IsStridedSliceSupported(const TensorInfo& input,
431  const TensorInfo& output,
432  const StridedSliceDescriptor& descriptor,
433  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
434 
435  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
436  bool IsSubtractionSupported(const TensorInfo& input0,
437  const TensorInfo& input1,
438  const TensorInfo& output,
439  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
440 
441  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
442  bool IsSwitchSupported(const TensorInfo& input0,
443  const TensorInfo& input1,
444  const TensorInfo& output0,
445  const TensorInfo& output1,
446  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
447 
448  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
450  const TensorInfo& input,
451  const TensorInfo& output,
452  const TransposeConvolution2dDescriptor& descriptor,
453  const TensorInfo& weights,
454  const Optional<TensorInfo>& biases,
455  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
456 
457  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
458  bool IsTransposeSupported(const TensorInfo& input,
459  const TensorInfo& output,
460  const TransposeDescriptor& descriptor,
461  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
462 
463  ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. Use IsLayerSupported instead.", "23.08")
465  const TensorInfo& input,
466  const TensorInfo& outputStateIn,
467  const TensorInfo& cellStateIn,
468  const TensorInfo& outputStateOut,
469  const TensorInfo& cellStateOut,
470  const TensorInfo& output,
471  const LstmDescriptor& descriptor,
472  const LstmInputParamsInfo& paramsInfo,
473  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
474 
475 };
476 
477 } // namespace armnn
bool IsSplitterSupported(const TensorInfo &input, const std::vector< std::reference_wrapper< TensorInfo >> &outputs, const ViewsDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsInputSupported(const TensorInfo &input, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A ViewsDescriptor for the SplitterLayer.
const TensorInfo const TensorInfo & anchors
bool IsLayerSupported(const LayerType &type, const std::vector< TensorInfo > &infos, const BaseDescriptor &descriptor, const Optional< LstmInputParamsInfo > &lstmParamsInfo=EmptyOptional(), const Optional< QuantizedLstmInputParamsInfo > &quantizedLstmParamsInfo=EmptyOptional(), Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsQLstmSupported(const TensorInfo &input, const TensorInfo &previousOutputIn, const TensorInfo &previousCellStateIn, const TensorInfo &outputStateOut, const TensorInfo &cellStateOut, const TensorInfo &output, const QLstmDescriptor &descriptor, const LstmInputParamsInfo &paramsInfo, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A TransposeConvolution2dDescriptor for the TransposeConvolution2dLayer.
const TensorInfo const TensorInfo const TensorInfo const TensorInfo const TensorInfo const TensorInfo const LstmDescriptor const LstmInputParamsInfo & paramsInfo
bool IsPreluSupported(const TensorInfo &input, const TensorInfo &alpha, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported) const override
const TensorInfo & output
A ReshapeDescriptor for the ReshapeLayer.
bool IsDetectionPostProcessSupported(const TensorInfo &boxEncodings, const TensorInfo &scores, const TensorInfo &anchors, const TensorInfo &detectionBoxes, const TensorInfo &detectionClasses, const TensorInfo &detectionScores, const TensorInfo &numDetections, const DetectionPostProcessDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsMemImportSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsLogSoftmaxSupported(const TensorInfo &input, const TensorInfo &output, const LogSoftmaxDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A ComparisonDescriptor for the ComparisonLayer.
Definition: Descriptors.hpp:89
bool IsTransposeSupported(const TensorInfo &input, const TensorInfo &output, const TransposeDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo const TensorInfo const TensorInfo const TensorInfo const TensorInfo & gamma
bool IsCastSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsQuantizeSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsMultiplicationSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsQuantizedLstmSupported(const TensorInfo &input, const TensorInfo &previousCellStateIn, const TensorInfo &previousOutputIn, const TensorInfo &cellStateOut, const TensorInfo &output, const QuantizedLstmInputParamsInfo &paramsInfo, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const std::vector< std::reference_wrapper< TensorInfo > > & outputs
bool IsLogicalBinarySupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, const LogicalBinaryDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A Convolution2dDescriptor for the Convolution2dLayer.
bool IsAdditionSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsDebugSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsActivationSupported(const TensorInfo &input, const TensorInfo &output, const ActivationDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsReduceSupported(const TensorInfo &input, const TensorInfo &output, const ReduceDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsMeanSupported(const TensorInfo &input, const TensorInfo &output, const MeanDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsArgMinMaxSupported(const TensorInfo &input, const TensorInfo &output, const ArgMinMaxDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo const ActivationDescriptor Optional< std::string & > reasonIfUnsupported
bool IsShapeSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A LogicalBinaryDescriptor for the LogicalBinaryLayer.
bool IsSpaceToBatchNdSupported(const TensorInfo &input, const TensorInfo &output, const SpaceToBatchNdDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo & scores
const TensorInfo const TensorInfo & output0
bool IsSoftmaxSupported(const TensorInfo &input, const TensorInfo &output, const SoftmaxDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsComparisonSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, const ComparisonDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo const TensorInfo const TensorInfo const TensorInfo & detectionClasses
Copyright (c) 2021 ARM Limited and Contributors.
bool IsStandInSupported(const std::vector< const TensorInfo *> &inputs, const std::vector< const TensorInfo *> &outputs, const StandInDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo const ActivationDescriptor & descriptor
bool IsDepthwiseConvolutionSupported(const TensorInfo &input, const TensorInfo &output, const DepthwiseConvolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsL2NormalizationSupported(const TensorInfo &input, const TensorInfo &output, const L2NormalizationDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A SpaceToDepthDescriptor for the SpaceToDepthLayer.
bool IsSliceSupported(const TensorInfo &input, const TensorInfo &output, const SliceDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsFakeQuantizationSupported(const TensorInfo &input, const FakeQuantizationDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer.
virtual bool IsFillSupported(const TensorInfo &input, const TensorInfo &output, const FillDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsPreCompiledSupported(const TensorInfo &input, const PreCompiledDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo & outputStateIn
const TensorInfo const TensorInfo & previousCellStateIn
const TensorInfo const TensorInfo const TensorInfo & output1
const TensorInfo const TensorInfo const TensorInfo const TensorInfo const TensorInfo const TensorInfo & numDetections
bool IsStridedSliceSupported(const TensorInfo &input, const TensorInfo &output, const StridedSliceDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A ResizeBilinearDescriptor for the ResizeBilinearLayer.
const TensorInfo & alpha
Base class for all descriptors.
Definition: Descriptors.hpp:22
A StackDescriptor for the StackLayer.
A PadDescriptor for the PadLayer.
bool IsSwitchSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output0, const TensorInfo &output1, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsBatchToSpaceNdSupported(const TensorInfo &input, const TensorInfo &output, const BatchToSpaceNdDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsChannelShuffleSupported(const TensorInfo &input, const TensorInfo &output, const ChannelShuffleDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsDivisionSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo const TensorInfo & cellStateIn
An LstmDescriptor for the LstmLayer.
A L2NormalizationDescriptor for the L2NormalizationLayer.
const TensorInfo const TensorInfo const TensorInfo & var
An ArgMinMaxDescriptor for ArgMinMaxLayer.
Definition: Descriptors.hpp:67
An OriginsDescriptor for the ConcatLayer.
A ReduceDescriptor for the REDUCE operators.
bool IsPermuteSupported(const TensorInfo &input, const TensorInfo &output, const PermuteDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A FullyConnectedDescriptor for the FullyConnectedLayer.
bool IsLogicalUnarySupported(const TensorInfo &input, const TensorInfo &output, const ElementwiseUnaryDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo const TensorInfo const TensorInfo const TensorInfo & outputStateOut
A FakeQuantizationDescriptor for the FakeQuantizationLayer.
bool IsConvertFp32ToFp16Supported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsElementwiseUnarySupported(const TensorInfo &input, const TensorInfo &output, const ElementwiseUnaryDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsConvertFp16ToFp32Supported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo const TensorInfo const TensorInfo const TensorInfo const TensorInfo & cellStateOut
A GatherDescriptor for the GatherLayer.
const TensorInfo const TensorInfo const TensorInfo const TensorInfo & beta
A StandInDescriptor for the StandIn layer.
bool IsReshapeSupported(const TensorInfo &input, const TensorInfo &output, const ReshapeDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A QLstmDescriptor for the QLstmLayer.
bool IsMergeSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsStackSupported(const std::vector< const TensorInfo *> &inputs, const TensorInfo &output, const StackDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsSpaceToDepthSupported(const TensorInfo &input, const TensorInfo &output, const SpaceToDepthDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
An ActivationDescriptor for the ActivationLayer.
Definition: Descriptors.hpp:36
bool IsMemCopySupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsInstanceNormalizationSupported(const TensorInfo &input, const TensorInfo &output, const InstanceNormalizationDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A SliceDescriptor for the SliceLayer.
bool IsRankSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported) const override
A Convolution3dDescriptor for the Convolution3dLayer.
bool IsTransposeConvolution2dSupported(const TensorInfo &input, const TensorInfo &output, const TransposeConvolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo & previousOutputIn
A Pooling3dDescriptor for the Pooling3dLayer.
A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer.
bool IsPadSupported(const TensorInfo &input, const TensorInfo &output, const PadDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsMinimumSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsConvertBf16ToFp32Supported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo const TensorInfo const TensorInfo const TensorInfo const TensorInfo & detectionScores
ARMNN_DEPRECATED_MSG_REMOVAL_DATE("This method is deprecated. " "Use ABI Stable IsLayerSupported accepting LayerType argument instead.", "23.08") virtual bool IsActivationSupported(const TensorInfo &input
EmptyOptional is used to initialize the Optional class in case we want to have default value for an O...
Definition: Optional.hpp:32
bool IsConvolution3dSupported(const TensorInfo &input, const TensorInfo &output, const Convolution3dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A ElementwiseUnaryDescriptor for the ElementwiseUnaryLayer.
bool IsConvolution2dSupported(const TensorInfo &input, const TensorInfo &output, const Convolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsSubtractionSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo const Convolution2dDescriptor const TensorInfo const Optional< TensorInfo > & biases
bool IsLstmSupported(const TensorInfo &input, const TensorInfo &outputStateIn, const TensorInfo &cellStateIn, const TensorInfo &scratchBuffer, const TensorInfo &outputStateOut, const TensorInfo &cellStateOut, const TensorInfo &output, const LstmDescriptor &descriptor, const LstmInputParamsInfo &paramsInfo, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A MeanDescriptor for the MeanLayer.
const TensorInfo const TensorInfo const TensorInfo & detectionBoxes
bool IsFloorSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsConstantSupported(const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsNormalizationSupported(const TensorInfo &input, const TensorInfo &output, const NormalizationDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A TransposeDescriptor for the TransposeLayer.
A StridedSliceDescriptor for the StridedSliceLayer.
bool IsFullyConnectedSupported(const TensorInfo &input, const TensorInfo &output, const TensorInfo &weights, const TensorInfo &biases, const FullyConnectedDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsOutputSupported(const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsResizeSupported(const TensorInfo &input, const TensorInfo &output, const ResizeDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsPooling2dSupported(const TensorInfo &input, const TensorInfo &output, const Pooling2dDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo & input1
A PreCompiledDescriptor for the PreCompiledLayer.
bool IsUnidirectionalSequenceLstmSupported(const TensorInfo &input, const TensorInfo &outputStateIn, const TensorInfo &cellStateIn, const TensorInfo &outputStateOut, const TensorInfo &cellStateOut, const TensorInfo &output, const LstmDescriptor &descriptor, const LstmInputParamsInfo &paramsInfo, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsGatherSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, const GatherDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A Pooling2dDescriptor for the Pooling2dLayer.
A NormalizationDescriptor for the NormalizationLayer.
bool IsConvertFp32ToBf16Supported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
bool IsMaximumSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo const TensorInfo const TensorInfo & scratchBuffer
An InstanceNormalizationDescriptor for InstanceNormalizationLayer.
bool IsPooling3dSupported(const TensorInfo &input, const TensorInfo &output, const Pooling3dDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A ChannelShuffleDescriptor for the ChannelShuffle operator.
bool IsDepthToSpaceSupported(const TensorInfo &input, const TensorInfo &output, const DepthToSpaceDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A SoftmaxDescriptor for the SoftmaxLayer.
bool IsConcatSupported(const std::vector< const TensorInfo *> inputs, const TensorInfo &output, const OriginsDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo const Convolution2dDescriptor const TensorInfo & weights
bool IsBatchNormalizationSupported(const TensorInfo &input, const TensorInfo &output, const TensorInfo &mean, const TensorInfo &var, const TensorInfo &beta, const TensorInfo &gamma, const BatchNormalizationDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
A DepthwiseConvolution2dDescriptor for the DepthwiseConvolution2dLayer.
A FillDescriptor for the FillLayer.
A BatchNormalizationDescriptor for the BatchNormalizationLayer.
bool IsDequantizeSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
const TensorInfo const TensorInfo & mean
A PermuteDescriptor for the PermuteLayer.
bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo &input, const TensorInfo &output, const DepthwiseConvolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
LayerType
When adding a new layer, adapt also the LastLayer enum value in the enum class LayerType below...
Definition: Types.hpp:468