ArmNN
 20.05
ILayerSupport.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #pragma once
6 
7 #include <armnn/Deprecated.hpp>
8 #include <armnn/DescriptorsFwd.hpp>
9 #include <armnn/LstmParams.hpp>
10 #include <armnn/Optional.hpp>
11 #include <armnn/QuantizedLstmParams.hpp>
12 
13 #include <cctype>
14 #include <functional>
15 #include <memory>
16 #include <vector>
17 
18 namespace armnn
19 {
20 
21 class TensorInfo;
22 
23 class ILayerSupport
24 {
25 protected:
26  ILayerSupport() {}
27  virtual ~ILayerSupport() {}
28 
29 public:
30  ARMNN_DEPRECATED_MSG("Use IsElementwiseUnarySupported instead")
31  virtual bool IsAbsSupported(const TensorInfo& input,
32  const TensorInfo& output,
33  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
34 
35  virtual bool IsActivationSupported(const TensorInfo& input,
36  const TensorInfo& output,
37  const ActivationDescriptor& descriptor,
38  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
39 
40  virtual bool IsAdditionSupported(const TensorInfo& input0,
41  const TensorInfo& input1,
42  const TensorInfo& output,
43  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
44 
45  virtual bool IsArgMinMaxSupported(const TensorInfo& input,
46  const TensorInfo& output,
47  const ArgMinMaxDescriptor& descriptor,
48  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
49 
50  virtual bool IsBatchNormalizationSupported(const TensorInfo& input,
51  const TensorInfo& output,
52  const TensorInfo& mean,
53  const TensorInfo& var,
54  const TensorInfo& beta,
55  const TensorInfo& gamma,
56  const BatchNormalizationDescriptor& descriptor,
57  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
58 
59  virtual bool IsBatchToSpaceNdSupported(const TensorInfo& input,
60  const TensorInfo& output,
61  const BatchToSpaceNdDescriptor& descriptor,
62  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
63 
64  virtual bool IsComparisonSupported(const TensorInfo& input0,
65  const TensorInfo& input1,
66  const TensorInfo& output,
67  const ComparisonDescriptor& descriptor,
68  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
69 
70  virtual bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,
71  const TensorInfo& output,
72  const OriginsDescriptor& descriptor,
73  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
74 
75  virtual bool IsConstantSupported(const TensorInfo& output,
76  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
77 
78  virtual bool IsConvertBf16ToFp32Supported(const TensorInfo& input,
79  const TensorInfo& output,
80  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
81 
82  virtual bool IsConvertFp32ToBf16Supported(const TensorInfo& input,
83  const TensorInfo& output,
84  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
85 
86  virtual bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
87  const TensorInfo& output,
88  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
89 
90  virtual bool IsConvertFp32ToFp16Supported(const TensorInfo& input,
91  const TensorInfo& output,
92  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
93 
94  virtual bool IsConvolution2dSupported(const TensorInfo& input,
95  const TensorInfo& output,
96  const Convolution2dDescriptor& descriptor,
97  const TensorInfo& weights,
98  const Optional<TensorInfo>& biases,
99  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
100 
101  virtual bool IsDebugSupported(const TensorInfo& input,
102  const TensorInfo& output,
103  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
104 
105  virtual bool IsDepthToSpaceSupported(const TensorInfo& input,
106  const TensorInfo& output,
107  const DepthToSpaceDescriptor& descriptor,
108  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
109 
110  virtual bool IsDepthwiseConvolutionSupported(
111  const TensorInfo& input,
112  const TensorInfo& output,
113  const DepthwiseConvolution2dDescriptor& descriptor,
114  const TensorInfo& weights,
115  const Optional<TensorInfo>& biases,
116  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
117 
118  virtual bool IsDequantizeSupported(const TensorInfo& input,
119  const TensorInfo& output,
120  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
121 
122  virtual bool IsDetectionPostProcessSupported(const TensorInfo& boxEncodings,
123  const TensorInfo& scores,
124  const TensorInfo& anchors,
125  const TensorInfo& detectionBoxes,
126  const TensorInfo& detectionClasses,
127  const TensorInfo& detectionScores,
128  const TensorInfo& numDetections,
129  const DetectionPostProcessDescriptor& descriptor,
130  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const =0;
131 
132  virtual bool IsDilatedDepthwiseConvolutionSupported(
133  const TensorInfo& input,
134  const TensorInfo& output,
135  const DepthwiseConvolution2dDescriptor& descriptor,
136  const TensorInfo& weights,
137  const Optional<TensorInfo>& biases,
138  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
139 
140  virtual bool IsDivisionSupported(const TensorInfo& input0,
141  const TensorInfo& input1,
142  const TensorInfo& output,
143  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
144 
145  virtual bool IsElementwiseUnarySupported(const TensorInfo& input,
146  const TensorInfo& output,
147  const ElementwiseUnaryDescriptor& descriptor,
148  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
149 
150  ARMNN_DEPRECATED_MSG("Use IsComparisonSupported instead")
151  virtual bool IsEqualSupported(const TensorInfo& input0,
152  const TensorInfo& input1,
153  const TensorInfo& output,
154  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
155 
156  virtual bool IsFakeQuantizationSupported(const TensorInfo& input,
157  const FakeQuantizationDescriptor& descriptor,
158  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
159 
160  virtual bool IsFloorSupported(const TensorInfo& input,
161  const TensorInfo& output,
162  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
163 
164  virtual bool IsFullyConnectedSupported(const TensorInfo& input,
165  const TensorInfo& output,
166  const TensorInfo& weights,
167  const TensorInfo& biases,
168  const FullyConnectedDescriptor& descriptor,
169  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
170 
171  virtual bool IsGatherSupported(const TensorInfo& input0,
172  const TensorInfo& input1,
173  const TensorInfo& output,
174  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
175 
176  ARMNN_DEPRECATED_MSG("Use IsComparisonSupported instead")
177  virtual bool IsGreaterSupported(const TensorInfo& input0,
178  const TensorInfo& input1,
179  const TensorInfo& ouput,
180  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
181 
182  virtual bool IsInputSupported(const TensorInfo& input,
183  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
184 
185  virtual bool IsInstanceNormalizationSupported(
186  const TensorInfo& input,
187  const TensorInfo& output,
188  const InstanceNormalizationDescriptor& descriptor,
189  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
190 
191  virtual bool IsL2NormalizationSupported(const TensorInfo& input,
192  const TensorInfo& output,
193  const L2NormalizationDescriptor& descriptor,
194  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
195 
196  virtual bool IsLogSoftmaxSupported(const TensorInfo& input,
197  const TensorInfo& output,
198  const LogSoftmaxDescriptor& descriptor,
199  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
200 
201  virtual bool IsLstmSupported(const TensorInfo& input,
202  const TensorInfo& outputStateIn,
203  const TensorInfo& cellStateIn,
204  const TensorInfo& scratchBuffer,
205  const TensorInfo& outputStateOut,
206  const TensorInfo& cellStateOut,
207  const TensorInfo& output,
208  const LstmDescriptor& descriptor,
209  const LstmInputParamsInfo& paramsInfo,
210  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
211 
212  virtual bool IsMaximumSupported(const TensorInfo& input0,
213  const TensorInfo& input1,
214  const TensorInfo& output,
215  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
216 
217  virtual bool IsMeanSupported(const TensorInfo& input,
218  const TensorInfo& output,
219  const MeanDescriptor& descriptor,
220  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
221 
222  virtual bool IsMemCopySupported(const TensorInfo& input,
223  const TensorInfo& output,
224  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
225 
226  virtual bool IsMemImportSupported(const TensorInfo& input,
227  const TensorInfo& output,
228  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
229 
230  virtual bool IsMergeSupported(const TensorInfo& input0,
231  const TensorInfo& input1,
232  const TensorInfo& output,
233  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
234 
235  ARMNN_DEPRECATED_MSG("Use IsConcatSupported instead")
236  virtual bool IsMergerSupported(const std::vector<const TensorInfo*> inputs,
237  const TensorInfo& output,
238  const OriginsDescriptor& descriptor,
239  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
240 
241  virtual bool IsMinimumSupported(const TensorInfo& input0,
242  const TensorInfo& input1,
243  const TensorInfo& ouput,
244  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
245 
246  virtual bool IsMultiplicationSupported(const TensorInfo& input0,
247  const TensorInfo& input1,
248  const TensorInfo& output,
249  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
250 
251  virtual bool IsNormalizationSupported(const TensorInfo& input,
252  const TensorInfo& output,
253  const NormalizationDescriptor& descriptor,
254  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
255 
256  virtual bool IsOutputSupported(const TensorInfo& output,
257  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
258 
259  virtual bool IsPadSupported(const TensorInfo& input,
260  const TensorInfo& output,
261  const PadDescriptor& descriptor,
262  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
263 
264  virtual bool IsPermuteSupported(const TensorInfo& input,
265  const TensorInfo& output,
266  const PermuteDescriptor& descriptor,
267  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
268 
269  virtual bool IsPooling2dSupported(const TensorInfo& input,
270  const TensorInfo& output,
271  const Pooling2dDescriptor& descriptor,
272  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
273 
274  virtual bool IsPreCompiledSupported(const TensorInfo& input,
275  const PreCompiledDescriptor& descriptor,
276  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
277 
278  virtual bool IsPreluSupported(const TensorInfo& input,
279  const TensorInfo& alpha,
280  const TensorInfo& output,
281  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
282 
283  virtual bool IsQuantizeSupported(const TensorInfo& input,
284  const TensorInfo& output,
285  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
286 
287  virtual bool IsQLstmSupported(const TensorInfo& input,
288  const TensorInfo& previousOutputIn,
289  const TensorInfo& previousCellStateIn,
290  const TensorInfo& outputStateOut,
291  const TensorInfo& cellStateOut,
292  const TensorInfo& output,
293  const QLstmDescriptor& descriptor,
294  const LstmInputParamsInfo& paramsInfo,
295  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
296 
297  virtual bool IsQuantizedLstmSupported(const TensorInfo& input,
298  const TensorInfo& previousCellStateIn,
299  const TensorInfo& previousOutputIn,
300  const TensorInfo& cellStateOut,
301  const TensorInfo& output,
302  const QuantizedLstmInputParamsInfo& paramsInfo,
303  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
304 
305  virtual bool IsReshapeSupported(const TensorInfo& input,
306  const TensorInfo& output,
307  const ReshapeDescriptor& descriptor,
308  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
309 
310  ARMNN_DEPRECATED_MSG("Use IsResizeSupported instead")
311  virtual bool IsResizeBilinearSupported(const TensorInfo& input,
312  const TensorInfo& output,
313  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
314 
315  virtual bool IsResizeSupported(const TensorInfo& input,
316  const TensorInfo& output,
317  const ResizeDescriptor& descriptor,
318  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
319 
320  ARMNN_DEPRECATED_MSG("Use IsElementwiseUnarySupported instead")
321  virtual bool IsRsqrtSupported(const TensorInfo& input,
322  const TensorInfo& output,
323  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
324 
325  virtual bool IsSliceSupported(const TensorInfo& input,
326  const TensorInfo& output,
327  const SliceDescriptor& descriptor,
328  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
329 
330  virtual bool IsSoftmaxSupported(const TensorInfo& input,
331  const TensorInfo& output,
332  const SoftmaxDescriptor& descriptor,
333  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
334 
335  virtual bool IsSpaceToBatchNdSupported(const TensorInfo& input,
336  const TensorInfo& output,
337  const SpaceToBatchNdDescriptor& descriptor,
338  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
339 
340  virtual bool IsSpaceToDepthSupported(const TensorInfo& input,
341  const TensorInfo& output,
342  const SpaceToDepthDescriptor& descriptor,
343  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
344 
345  ARMNN_DEPRECATED_MSG("Use IsSplitterSupported with outputs instead")
346  virtual bool IsSplitterSupported(const TensorInfo& input,
347  const ViewsDescriptor& descriptor,
348  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
349 
350  virtual bool IsSplitterSupported(const TensorInfo& input,
351  const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
352  const ViewsDescriptor& descriptor,
353  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
354 
355  virtual bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
356  const TensorInfo& output,
357  const StackDescriptor& descriptor,
358  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
359 
360  virtual bool IsStandInSupported(const std::vector<const TensorInfo*>& inputs,
361  const std::vector<const TensorInfo*>& outputs,
362  const StandInDescriptor& descriptor,
363  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
364 
365 
366  virtual bool IsStridedSliceSupported(const TensorInfo& input,
367  const TensorInfo& output,
368  const StridedSliceDescriptor& descriptor,
369  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
370 
371  virtual bool IsSubtractionSupported(const TensorInfo& input0,
372  const TensorInfo& input1,
373  const TensorInfo& output,
374  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
375 
376  virtual bool IsSwitchSupported(const TensorInfo& input0,
377  const TensorInfo& input1,
378  const TensorInfo& output0,
379  const TensorInfo& output1,
380  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
381 
382  virtual bool IsTransposeConvolution2dSupported(
383  const TensorInfo& input,
384  const TensorInfo& output,
385  const TransposeConvolution2dDescriptor& descriptor,
386  const TensorInfo& weights,
387  const Optional<TensorInfo>& biases,
388  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
389 
390  virtual bool IsTransposeSupported(const TensorInfo& input,
391  const TensorInfo& output,
392  const TransposeDescriptor& descriptor,
393  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0;
394 
395 }; // class ILayerSupport
396 
397 using ILayerSupportSharedPtr = std::shared_ptr<ILayerSupport>;
398 
399 } // namespace armnn
armnn::ILayerSupport::IsTransposeSupported
virtual bool IsTransposeSupported(const TensorInfo &input, const TensorInfo &output, const TransposeDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsConvolution2dSupported
virtual bool IsConvolution2dSupported(const TensorInfo &input, const TensorInfo &output, const Convolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsInstanceNormalizationSupported
virtual bool IsInstanceNormalizationSupported(const TensorInfo &input, const TensorInfo &output, const InstanceNormalizationDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ViewsDescriptor
A ViewsDescriptor for the SplitterLayer.
Definition: Descriptors.hpp:198
armnn::TransposeConvolution2dDescriptor
A TransposeConvolution2dDescriptor for the TransposeConvolution2dLayer.
Definition: Descriptors.hpp:1151
armnn::ReshapeDescriptor
A ReshapeDescriptor for the ReshapeLayer.
Definition: Descriptors.hpp:770
armnn::ILayerSupport::IsDepthToSpaceSupported
virtual bool IsDepthToSpaceSupported(const TensorInfo &input, const TensorInfo &output, const DepthToSpaceDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsResizeBilinearSupported
virtual bool IsResizeBilinearSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsMemImportSupported
virtual bool IsMemImportSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ComparisonDescriptor
A ComparisonDescriptor for the ComparisonLayer.
Definition: Descriptors.hpp:70
armnn::ILayerSupport::IsFloorSupported
virtual bool IsFloorSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::Convolution2dDescriptor
A Convolution2dDescriptor for the Convolution2dLayer.
Definition: Descriptors.hpp:400
armnn::ILayerSupport::IsConvertFp16ToFp32Supported
virtual bool IsConvertFp16ToFp32Supported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupportSharedPtr
std::shared_ptr< ILayerSupport > ILayerSupportSharedPtr
Definition: ILayerSupport.hpp:397
armnn::ILayerSupport::IsMergeSupported
virtual bool IsMergeSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
QuantizedLstmParams.hpp
armnn::ILayerSupport::IsTransposeConvolution2dSupported
virtual bool IsTransposeConvolution2dSupported(const TensorInfo &input, const TensorInfo &output, const TransposeConvolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsGreaterSupported
virtual bool IsGreaterSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &ouput, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsMergerSupported
virtual bool IsMergerSupported(const std::vector< const TensorInfo *> inputs, const TensorInfo &output, const OriginsDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsMaximumSupported
virtual bool IsMaximumSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsDebugSupported
virtual bool IsDebugSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsConcatSupported
virtual bool IsConcatSupported(const std::vector< const TensorInfo *> inputs, const TensorInfo &output, const OriginsDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
boxEncodings
std::vector< float > boxEncodings({ 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1.0f, 0.0f, 0.0f, 0.0f, -1.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f })
armnn
Copyright (c) 2020 ARM Limited.
Definition: 00_introduction.dox:25
LstmParams.hpp
armnn::ILayerSupport::IsSplitterSupported
virtual bool IsSplitterSupported(const TensorInfo &input, const ViewsDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::SpaceToDepthDescriptor
A SpaceToDepthDescriptor for the SpaceToDepthLayer.
Definition: Descriptors.hpp:822
armnn::ILayerSupport::IsGatherSupported
virtual bool IsGatherSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsMultiplicationSupported
virtual bool IsMultiplicationSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::BatchToSpaceNdDescriptor
A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer.
Definition: Descriptors.hpp:665
armnn::ILayerSupport::IsBatchToSpaceNdSupported
virtual bool IsBatchToSpaceNdSupported(const TensorInfo &input, const TensorInfo &output, const BatchToSpaceNdDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ResizeDescriptor
A ResizeDescriptor for the ResizeLayer.
Definition: Descriptors.hpp:732
armnn::StackDescriptor
A StackDescriptor for the StackLayer.
Definition: Descriptors.hpp:962
armnn::PadDescriptor
A PadDescriptor for the PadLayer.
Definition: Descriptors.hpp:913
armnn::ILayerSupport::IsLstmSupported
virtual 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 =0
armnn::ILayerSupport::IsElementwiseUnarySupported
virtual bool IsElementwiseUnarySupported(const TensorInfo &input, const TensorInfo &output, const ElementwiseUnaryDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsSoftmaxSupported
virtual bool IsSoftmaxSupported(const TensorInfo &input, const TensorInfo &output, const SoftmaxDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsSliceSupported
virtual bool IsSliceSupported(const TensorInfo &input, const TensorInfo &output, const SliceDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsStandInSupported
virtual bool IsStandInSupported(const std::vector< const TensorInfo *> &inputs, const std::vector< const TensorInfo *> &outputs, const StandInDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::LstmDescriptor
An LstmDescriptor for the LstmLayer.
Definition: Descriptors.hpp:849
armnn::ILayerSupport::IsPreCompiledSupported
virtual bool IsPreCompiledSupported(const TensorInfo &input, const PreCompiledDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsBatchNormalizationSupported
virtual 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 =0
Optional.hpp
armnn::ILayerSupport::IsPadSupported
virtual bool IsPadSupported(const TensorInfo &input, const TensorInfo &output, const PadDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::L2NormalizationDescriptor
A L2NormalizationDescriptor for the L2NormalizationLayer.
Definition: Descriptors.hpp:599
armnn::ArgMinMaxDescriptor
An ArgMinMaxDescriptor for ArgMinMaxLayer.
Definition: Descriptors.hpp:51
armnn::OriginsDescriptor
An OriginsDescriptor for the ConcatLayer.
Definition: Descriptors.hpp:155
armnn::FullyConnectedDescriptor
A FullyConnectedDescriptor for the FullyConnectedLayer.
Definition: Descriptors.hpp:381
armnn::ILayerSupport::IsPermuteSupported
virtual bool IsPermuteSupported(const TensorInfo &input, const TensorInfo &output, const PermuteDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsLogSoftmaxSupported
virtual bool IsLogSoftmaxSupported(const TensorInfo &input, const TensorInfo &output, const LogSoftmaxDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsMemCopySupported
virtual bool IsMemCopySupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsMeanSupported
virtual bool IsMeanSupported(const TensorInfo &input, const TensorInfo &output, const MeanDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsQLstmSupported
virtual 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 =0
armnn::ILayerSupport::IsInputSupported
virtual bool IsInputSupported(const TensorInfo &input, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsReshapeSupported
virtual bool IsReshapeSupported(const TensorInfo &input, const TensorInfo &output, const ReshapeDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::FakeQuantizationDescriptor
A FakeQuantizationDescriptor for the FakeQuantizationLayer.
Definition: Descriptors.hpp:696
armnn::ILayerSupport::IsActivationSupported
virtual bool IsActivationSupported(const TensorInfo &input, const TensorInfo &output, const ActivationDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsDepthwiseConvolutionSupported
virtual bool IsDepthwiseConvolutionSupported(const TensorInfo &input, const TensorInfo &output, const DepthwiseConvolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::StandInDescriptor
A StandInDescriptor for the StandIn layer.
Definition: Descriptors.hpp:992
armnn::QLstmDescriptor
A QLstmDescriptor for the QLstmLayer.
Definition: Descriptors.hpp:1091
armnn::ILayerSupport::IsDivisionSupported
virtual bool IsDivisionSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsL2NormalizationSupported
virtual bool IsL2NormalizationSupported(const TensorInfo &input, const TensorInfo &output, const L2NormalizationDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ActivationDescriptor
An ActivationDescriptor for the ActivationLayer.
Definition: Descriptors.hpp:20
armnn::ILayerSupport::IsQuantizeSupported
virtual bool IsQuantizeSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::SliceDescriptor
A SliceDescriptor for the SliceLayer.
Definition: Descriptors.hpp:939
armnn::ILayerSupport::IsRsqrtSupported
virtual bool IsRsqrtSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsPooling2dSupported
virtual bool IsPooling2dSupported(const TensorInfo &input, const TensorInfo &output, const Pooling2dDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsArgMinMaxSupported
virtual bool IsArgMinMaxSupported(const TensorInfo &input, const TensorInfo &output, const ArgMinMaxDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsPreluSupported
virtual bool IsPreluSupported(const TensorInfo &input, const TensorInfo &alpha, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::SpaceToBatchNdDescriptor
A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer.
Definition: Descriptors.hpp:790
Deprecated.hpp
armnn::ILayerSupport::IsStackSupported
virtual bool IsStackSupported(const std::vector< const TensorInfo *> &inputs, const TensorInfo &output, const StackDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::EmptyOptional
EmptyOptional is used to initialize the Optional class in case we want to have default value for an O...
Definition: Optional.hpp:32
armnn::ElementwiseUnaryDescriptor
A ElementwiseUnaryDescriptor for the ElementwiseUnaryLayer.
Definition: Descriptors.hpp:90
scores
std::vector< float > scores({ 0.0f, 0.9f, 0.8f, 0.0f, 0.75f, 0.72f, 0.0f, 0.6f, 0.5f, 0.0f, 0.93f, 0.95f, 0.0f, 0.5f, 0.4f, 0.0f, 0.3f, 0.2f })
armnn::ILayerSupport::IsFakeQuantizationSupported
virtual bool IsFakeQuantizationSupported(const TensorInfo &input, const FakeQuantizationDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsAbsSupported
virtual bool IsAbsSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsStridedSliceSupported
virtual bool IsStridedSliceSupported(const TensorInfo &input, const TensorInfo &output, const StridedSliceDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsQuantizedLstmSupported
virtual 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 =0
armnn::ILayerSupport::IsAdditionSupported
virtual bool IsAdditionSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::MeanDescriptor
A MeanDescriptor for the MeanLayer.
Definition: Descriptors.hpp:889
armnn::ILayerSupport::IsNormalizationSupported
virtual bool IsNormalizationSupported(const TensorInfo &input, const TensorInfo &output, const NormalizationDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsEqualSupported
virtual bool IsEqualSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::TransposeDescriptor
A TransposeDescriptor for the TransposeLayer.
Definition: Descriptors.hpp:1195
armnn::StridedSliceDescriptor
A StridedSliceDescriptor for the StridedSliceLayer.
Definition: Descriptors.hpp:1014
armnn::ILayerSupport::IsSwitchSupported
virtual bool IsSwitchSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output0, const TensorInfo &output1, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsConvertFp32ToFp16Supported
virtual bool IsConvertFp32ToFp16Supported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsConvertFp32ToBf16Supported
virtual bool IsConvertFp32ToBf16Supported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
ARMNN_DEPRECATED_MSG
#define ARMNN_DEPRECATED_MSG(message)
Definition: Deprecated.hpp:43
armnn::ILayerSupport::IsConvertBf16ToFp32Supported
virtual bool IsConvertBf16ToFp32Supported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::PreCompiledDescriptor
A PreCompiledDescriptor for the PreCompiledLayer.
Definition: Descriptors.hpp:1078
armnn::ILayerSupport::IsSpaceToBatchNdSupported
virtual bool IsSpaceToBatchNdSupported(const TensorInfo &input, const TensorInfo &output, const SpaceToBatchNdDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
DescriptorsFwd.hpp
armnn::Pooling2dDescriptor
A Pooling2dDescriptor for the Pooling2dLayer.
Definition: Descriptors.hpp:321
armnn::NormalizationDescriptor
A NormalizationDescriptor for the NormalizationLayer.
Definition: Descriptors.hpp:559
armnn::ILayerSupport::IsSubtractionSupported
virtual bool IsSubtractionSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::DetectionPostProcessDescriptor
Definition: Descriptors.hpp:503
armnn::InstanceNormalizationDescriptor
An InstanceNormalizationDescriptor for InstanceNormalizationLayer.
Definition: Descriptors.hpp:637
armnn::ILayerSupport::IsFullyConnectedSupported
virtual bool IsFullyConnectedSupported(const TensorInfo &input, const TensorInfo &output, const TensorInfo &weights, const TensorInfo &biases, const FullyConnectedDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsDequantizeSupported
virtual bool IsDequantizeSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsOutputSupported
virtual bool IsOutputSupported(const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsResizeSupported
virtual bool IsResizeSupported(const TensorInfo &input, const TensorInfo &output, const ResizeDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::SoftmaxDescriptor
A SoftmaxDescriptor for the SoftmaxLayer.
Definition: Descriptors.hpp:131
armnn::ILayerSupport::IsConstantSupported
virtual bool IsConstantSupported(const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsComparisonSupported
virtual bool IsComparisonSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, const ComparisonDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsDilatedDepthwiseConvolutionSupported
virtual bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo &input, const TensorInfo &output, const DepthwiseConvolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::DepthwiseConvolution2dDescriptor
A DepthwiseConvolution2dDescriptor for the DepthwiseConvolution2dLayer.
Definition: Descriptors.hpp:452
armnn::BatchNormalizationDescriptor
A BatchNormalizationDescriptor for the BatchNormalizationLayer.
Definition: Descriptors.hpp:618
armnn::ILayerSupport::IsMinimumSupported
virtual bool IsMinimumSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &ouput, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0
armnn::ILayerSupport::IsDetectionPostProcessSupported
virtual 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 =0
armnn::PermuteDescriptor
A PermuteDescriptor for the PermuteLayer.
Definition: Descriptors.hpp:110
anchors
std::vector< float > anchors({ 0.5f, 0.5f, 1.0f, 1.0f, 0.5f, 0.5f, 1.0f, 1.0f, 0.5f, 0.5f, 1.0f, 1.0f, 0.5f, 10.5f, 1.0f, 1.0f, 0.5f, 10.5f, 1.0f, 1.0f, 0.5f, 100.5f, 1.0f, 1.0f })
armnn::ILayerSupport::IsSpaceToDepthSupported
virtual bool IsSpaceToDepthSupported(const TensorInfo &input, const TensorInfo &output, const SpaceToDepthDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const =0