ArmNN
 24.08
NeonLayerSupport.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017-2023 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #pragma once
6 
8 
10 
11 namespace armnn
12 {
13 
15 {
16 public:
19 
21 
22  bool IsLayerSupported(const LayerType& type,
23  const std::vector<TensorInfo>& infos,
24  const BaseDescriptor& descriptor,
25  const Optional<LstmInputParamsInfo>& lstmParamsInfo,
26  const Optional<QuantizedLstmInputParamsInfo>& quantizedLstmParamsInfo,
27  Optional<std::string&> reasonIfUnsupported) const override;
28 
29  bool IsActivationSupported(const TensorInfo& input,
30  const TensorInfo& output,
31  const ActivationDescriptor& descriptor,
32  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
33 
34  bool IsAdditionSupported(const TensorInfo& input0,
35  const TensorInfo& input1,
36  const TensorInfo& output,
37  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
38 
39  bool IsArgMinMaxSupported(const TensorInfo& input,
40  const TensorInfo& output,
41  const ArgMinMaxDescriptor& descriptor,
42  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
43 
44  bool IsBatchMatMulSupported(const TensorInfo& inputX,
45  const TensorInfo& inputY,
46  const TensorInfo& output,
47  const BatchMatMulDescriptor& descriptor,
48  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
49 
50  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;
58 
59  bool IsBatchToSpaceNdSupported(const TensorInfo& input,
60  const TensorInfo& output,
61  const BatchToSpaceNdDescriptor& descriptor,
62  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
63 
64  bool IsCastSupported(const TensorInfo& input,
65  const TensorInfo& output,
66  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
67 
68  bool IsChannelShuffleSupported(const TensorInfo& input,
69  const TensorInfo& output,
70  const ChannelShuffleDescriptor& descriptor,
71  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
72 
73  bool IsComparisonSupported(const TensorInfo& input0,
74  const TensorInfo& input1,
75  const TensorInfo& output,
76  const ComparisonDescriptor& descriptor,
77  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
78 
79  bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,
80  const TensorInfo& output,
81  const OriginsDescriptor& descriptor,
82  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
83 
84  bool IsConstantSupported(const TensorInfo& output,
85  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
86 
87  bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
88  const TensorInfo& output,
89  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
90 
91  bool IsConvertFp32ToFp16Supported(const TensorInfo& input,
92  const TensorInfo& output,
93  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
94 
95  bool IsConvolution2dSupported(const TensorInfo& input,
96  const TensorInfo& output,
97  const Convolution2dDescriptor& descriptor,
98  const TensorInfo& weights,
99  const Optional<TensorInfo>& biases,
100  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
101 
102  bool IsConvolution3dSupported(const TensorInfo& input,
103  const TensorInfo& output,
104  const Convolution3dDescriptor& descriptor,
105  const TensorInfo& weights,
106  const Optional<TensorInfo>& biases,
107  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
108 
109  bool IsDepthToSpaceSupported(const TensorInfo& input,
110  const TensorInfo& output,
111  const DepthToSpaceDescriptor& descriptor,
112  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
113 
114 
116  const TensorInfo& output,
117  const DepthwiseConvolution2dDescriptor& descriptor,
118  const TensorInfo& weights,
119  const Optional<TensorInfo>& biases,
120  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
121 
122  bool IsDequantizeSupported(const TensorInfo& input,
123  const TensorInfo& output,
124  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
125 
127  const TensorInfo& output,
128  const DepthwiseConvolution2dDescriptor& descriptor,
129  const TensorInfo& weights,
130  const Optional<TensorInfo>& biases,
131  Optional<std::string&> reason = EmptyOptional()) const;
132 
133  bool IsElementwiseUnarySupported(const TensorInfo& input,
134  const TensorInfo& output,
135  const ElementwiseUnaryDescriptor& descriptor,
136  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
137 
138  bool IsFillSupported(const TensorInfo& input,
139  const TensorInfo& output,
140  const FillDescriptor& descriptor,
141  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
142 
143  bool IsFloorSupported(const TensorInfo& input,
144  const TensorInfo& output,
145  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
146 
147  bool IsFullyConnectedSupported(const TensorInfo& input,
148  const TensorInfo& output,
149  const TensorInfo& weights,
150  const TensorInfo& biases,
151  const FullyConnectedDescriptor& descriptor,
152  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
153 
154  bool IsFusedSupported(const std::vector<std::reference_wrapper<TensorInfo>>& inputs,
155  const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
156  const FusedDescriptor& descriptor,
157  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
158 
159  bool IsGatherNdSupported(const TensorInfo& input0,
160  const TensorInfo& input1,
161  const TensorInfo& output,
162  Optional<std::string&> reasonIfUnsupported) const;
163 
164  bool IsGatherSupported(const TensorInfo& input0,
165  const TensorInfo& input1,
166  const TensorInfo& output,
167  const GatherDescriptor& descriptor,
168  Optional<std::string&> reasonIfUnsupported) const;
169 
170  bool IsInputSupported(const TensorInfo& input,
171  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
172 
174  const TensorInfo& output,
175  const InstanceNormalizationDescriptor& descriptor,
176  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
177 
178  bool IsL2NormalizationSupported(const TensorInfo& input,
179  const TensorInfo& output,
180  const L2NormalizationDescriptor& descriptor,
181  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
182 
183  bool IsLogicalBinarySupported(const TensorInfo& input0,
184  const TensorInfo& input1,
185  const TensorInfo& output,
186  const LogicalBinaryDescriptor& descriptor,
187  Optional<std::string&> reasonIfUnsupported) const;
188 
189  bool IsLogSoftmaxSupported(const TensorInfo& input,
190  const TensorInfo& output,
191  const LogSoftmaxDescriptor& descriptor,
192  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
193 
194  bool IsLstmSupported(const TensorInfo& input,
195  const TensorInfo& outputStateIn,
196  const TensorInfo& cellStateIn,
197  const TensorInfo& scratchBuffer,
198  const TensorInfo& outputStateOut,
199  const TensorInfo& cellStateOut,
200  const TensorInfo& output,
201  const LstmDescriptor& descriptor,
202  const LstmInputParamsInfo& paramsInfo,
203  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
204 
205  bool IsMaximumSupported(const TensorInfo& input0,
206  const TensorInfo& input1,
207  const TensorInfo& output,
208  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
209 
210  bool IsMeanSupported(const TensorInfo& input,
211  const TensorInfo& output,
212  const MeanDescriptor& descriptor,
213  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
214 
215  bool IsMinimumSupported(const TensorInfo& input0,
216  const TensorInfo& input1,
217  const TensorInfo& output,
218  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
219 
220  bool IsMultiplicationSupported(const TensorInfo& input0,
221  const TensorInfo& input1,
222  const TensorInfo& output,
223  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
224 
225  bool IsDivisionSupported(const TensorInfo& input0,
226  const TensorInfo& input1,
227  const TensorInfo& output,
228  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
229 
230  bool IsNormalizationSupported(const TensorInfo& input,
231  const TensorInfo& output,
232  const NormalizationDescriptor& descriptor,
233  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
234 
235  bool IsOutputSupported(const TensorInfo& output,
236  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
237 
238  bool IsPadSupported(const TensorInfo& input,
239  const TensorInfo& output,
240  const PadDescriptor& descriptor,
241  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
242 
243  bool IsPermuteSupported(const TensorInfo& input,
244  const TensorInfo& output,
245  const PermuteDescriptor& descriptor,
246  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
247 
248  bool IsPooling2dSupported(const TensorInfo& input,
249  const TensorInfo& output,
250  const Pooling2dDescriptor& descriptor,
251  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
252  bool IsPooling3dSupported(const TensorInfo& input,
253  const TensorInfo& output,
254  const Pooling3dDescriptor& descriptor,
255  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
256  bool IsPreluSupported(const TensorInfo& input,
257  const TensorInfo& alpha,
258  const TensorInfo& output,
259  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
260 
261  bool IsQLstmSupported(const TensorInfo& input,
262  const TensorInfo& previousOutputIn,
263  const TensorInfo& previousCellStateIn,
264  const TensorInfo& outputStateOut,
265  const TensorInfo& cellStateOut,
266  const TensorInfo& output,
267  const QLstmDescriptor& descriptor,
268  const LstmInputParamsInfo& paramsInfo,
269  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
270 
271  bool IsQuantizeSupported(const TensorInfo& input,
272  const TensorInfo& output,
273  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
274 
275  bool IsQuantizedLstmSupported(const TensorInfo& input,
276  const TensorInfo& cellStateIn,
277  const TensorInfo& outputStateIn,
278  const TensorInfo& cellStateOut,
279  const TensorInfo& outputStateOut,
280  const QuantizedLstmInputParamsInfo& paramsInfo,
281  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
282 
283  bool IsReduceSupported(const TensorInfo& input,
284  const TensorInfo& output,
285  const ReduceDescriptor& descriptor,
286  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
287 
288  bool IsReshapeSupported(const TensorInfo& input,
289  const TensorInfo& output,
290  const ReshapeDescriptor& descriptor,
291  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
292 
293  bool IsResizeSupported(const TensorInfo& input,
294  const TensorInfo& output,
295  const ResizeDescriptor& descriptor,
296  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
297 
298  bool IsReverseV2Supported(const TensorInfo& input,
299  const TensorInfo& axis,
300  const TensorInfo& output,
301  Optional<std::string&> reasonIfUnsupported) const;
302 
303  bool IsSliceSupported(const TensorInfo& input,
304  const TensorInfo& output,
305  const SliceDescriptor& descriptor,
306  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
307 
308  bool IsSoftmaxSupported(const TensorInfo& input,
309  const TensorInfo& output,
310  const SoftmaxDescriptor& descriptor,
311  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
312 
313  bool IsSpaceToBatchNdSupported(const TensorInfo& input,
314  const TensorInfo& output,
315  const SpaceToBatchNdDescriptor& descriptor,
316  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
317 
318  bool IsSpaceToDepthSupported(const TensorInfo& input,
319  const TensorInfo& output,
320  const SpaceToDepthDescriptor& descriptor,
321  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
322 
323  bool IsSplitterSupported(const TensorInfo& input,
324  const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
325  const ViewsDescriptor& descriptor,
326  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
327 
328  bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
329  const TensorInfo& output,
330  const StackDescriptor& descriptor,
331  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
332 
333  bool IsStridedSliceSupported(const TensorInfo& input,
334  const TensorInfo& output,
335  const StridedSliceDescriptor& descriptor,
336  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
337 
338  bool IsSubtractionSupported(const TensorInfo& input0,
339  const TensorInfo& input1,
340  const TensorInfo& output,
341  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
342 
343  bool IsTileSupported(const TensorInfo& input0,
344  const TensorInfo& output,
345  const TileDescriptor& descriptor,
346  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
347 
349  const TensorInfo& output,
350  const TransposeConvolution2dDescriptor& descriptor,
351  const TensorInfo& weights,
352  const Optional<TensorInfo>& biases,
353  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
354 
355  bool IsTransposeSupported(const TensorInfo& input,
356  const TensorInfo& output,
357  const TransposeDescriptor& descriptor,
358  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
359 
361  const TensorInfo& outputStateIn,
362  const TensorInfo& cellStateIn,
363  const TensorInfo& outputStateOut,
364  const TensorInfo& cellStateOut,
365  const TensorInfo& output,
366  const UnidirectionalSequenceLstmDescriptor& descriptor,
367  const LstmInputParamsInfo& paramsInfo,
368  Optional<std::string&> reasonIfUnsupported) const;
369 
370 private:
372 
373 }; // class NeonLayerSupport
374 
375 } // namespace armnn
armnn::BatchNormalizationDescriptor
A BatchNormalizationDescriptor for the BatchNormalizationLayer.
Definition: Descriptors.hpp:828
armnn::NeonLayerSupport::IsAdditionSupported
bool IsAdditionSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:783
armnn::NeonLayerSupport::IsDequantizeSupported
bool IsDequantizeSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1070
armnn::ViewsDescriptor
A ViewsDescriptor for the SplitterLayer.
Definition: Descriptors.hpp:244
armnn::ActivationDescriptor
An ActivationDescriptor for the ActivationLayer.
Definition: Descriptors.hpp:36
armnn::FullyConnectedDescriptor
A FullyConnectedDescriptor for the FullyConnectedLayer.
Definition: Descriptors.hpp:507
armnn::NeonLayerSupport::IsSpaceToBatchNdSupported
bool IsSpaceToBatchNdSupported(const TensorInfo &input, const TensorInfo &output, const SpaceToBatchNdDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1570
armnn::QLstmDescriptor
A QLstmDescriptor for the QLstmLayer.
Definition: Descriptors.hpp:1380
armnn::Optional
Definition: Optional.hpp:270
armnn::NeonLayerSupport::IsPermuteSupported
bool IsPermuteSupported(const TensorInfo &input, const TensorInfo &output, const PermuteDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1407
armnn::NeonLayerSupport::IsStackSupported
bool IsStackSupported(const std::vector< const TensorInfo * > &inputs, const TensorInfo &output, const StackDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1627
armnn::Pooling3dDescriptor
A Pooling3dDescriptor for the Pooling3dLayer.
Definition: Descriptors.hpp:431
armnn::ResizeDescriptor
A ResizeDescriptor for the ResizeLayer.
Definition: Descriptors.hpp:985
armnn::ArgMinMaxDescriptor
An ArgMinMaxDescriptor for ArgMinMaxLayer.
Definition: Descriptors.hpp:67
armnn::InstanceNormalizationDescriptor
An InstanceNormalizationDescriptor for InstanceNormalizationLayer.
Definition: Descriptors.hpp:847
armnn::GatherDescriptor
A GatherDescriptor for the GatherLayer.
Definition: Descriptors.hpp:965
armnn::NeonLayerSupport
Definition: NeonLayerSupport.hpp:14
armnn::NeonLayerSupport::IsBatchMatMulSupported
bool IsBatchMatMulSupported(const TensorInfo &inputX, const TensorInfo &inputY, const TensorInfo &output, const BatchMatMulDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:808
armnn::NeonLayerSupport::IsPooling3dSupported
bool IsPooling3dSupported(const TensorInfo &input, const TensorInfo &output, const Pooling3dDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1423
armnn::NeonLayerSupport::IsConvertFp32ToFp16Supported
bool IsConvertFp32ToFp16Supported(const TensorInfo &input, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:965
armnn::TensorInfo
Definition: Tensor.hpp:152
armnn::NeonLayerSupport::IsPreluSupported
bool IsPreluSupported(const TensorInfo &input, const TensorInfo &alpha, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1431
armnn::L2NormalizationDescriptor
A L2NormalizationDescriptor for the L2NormalizationLayer.
Definition: Descriptors.hpp:809
armnn::NormalizationDescriptor
A NormalizationDescriptor for the NormalizationLayer.
Definition: Descriptors.hpp:769
armnn::ChannelShuffleDescriptor
A ChannelShuffleDescriptor for the ChannelShuffle operator.
Definition: Descriptors.hpp:1562
armnn::NeonLayerSupport::IsMultiplicationSupported
bool IsMultiplicationSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1351
armnn::NeonLayerSupport::IsFusedSupported
bool IsFusedSupported(const std::vector< std::reference_wrapper< TensorInfo >> &inputs, const std::vector< std::reference_wrapper< TensorInfo >> &outputs, const FusedDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1193
armnn::StackDescriptor
A StackDescriptor for the StackLayer.
Definition: Descriptors.hpp:1251
armnn::NeonLayerSupport::IsConstantSupported
bool IsConstantSupported(const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:947
armnn::NeonLayerSupport::IsLogicalBinarySupported
bool IsLogicalBinarySupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, const LogicalBinaryDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported) const
Definition: NeonLayerSupport.cpp:1258
armnn::NeonLayerSupport::IsBatchToSpaceNdSupported
bool IsBatchToSpaceNdSupported(const TensorInfo &input, const TensorInfo &output, const BatchToSpaceNdDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:859
armnn::NeonLayerSupport::IsReshapeSupported
bool IsReshapeSupported(const TensorInfo &input, const TensorInfo &output, const ReshapeDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1514
armnn::QuantizedLstmInputParamsInfo
Definition: QuantizedLstmParams.hpp:119
armnn::NeonLayerSupport::IsBatchNormalizationSupported
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
Definition: NeonLayerSupport.cpp:838
armnn::NeonLayerSupport::IsGatherNdSupported
bool IsGatherNdSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported) const
Definition: NeonLayerSupport.cpp:1220
armnn::NeonLayerSupport::IsSoftmaxSupported
bool IsSoftmaxSupported(const TensorInfo &input, const TensorInfo &output, const SoftmaxDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1562
armnn::NeonLayerSupport::IsSpaceToDepthSupported
bool IsSpaceToDepthSupported(const TensorInfo &input, const TensorInfo &output, const SpaceToDepthDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1582
LayerSupportBase.hpp
armnn::NeonLayerSupport::IsLayerSupported
bool IsLayerSupported(const LayerType &type, const std::vector< TensorInfo > &infos, const BaseDescriptor &descriptor, const Optional< LstmInputParamsInfo > &lstmParamsInfo, const Optional< QuantizedLstmInputParamsInfo > &quantizedLstmParamsInfo, Optional< std::string & > reasonIfUnsupported) const override
Default implementation of the ILayerSupport interface, Backends should implement this as a switch sta...
Definition: NeonLayerSupport.cpp:722
armnn::FusedDescriptor
A FusedDescriptor for the FusedLayer.
Definition: Descriptors.hpp:944
armnn::NeonLayerSupport::IsResizeSupported
bool IsResizeSupported(const TensorInfo &input, const TensorInfo &output, const ResizeDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1526
armnn::NeonLayerSupport::IsReduceSupported
bool IsReduceSupported(const TensorInfo &input, const TensorInfo &output, const ReduceDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1502
armnn::NeonLayerSupport::IsNormalizationSupported
bool IsNormalizationSupported(const TensorInfo &input, const TensorInfo &output, const NormalizationDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1377
armnn::NeonLayerSupport::IsSliceSupported
bool IsSliceSupported(const TensorInfo &input, const TensorInfo &output, const SliceDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1550
armnn::NeonLayerSupport::IsSubtractionSupported
bool IsSubtractionSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1651
armnn::PadDescriptor
A PadDescriptor for the PadLayer.
Definition: Descriptors.hpp:1196
armnn::NeonLayerSupport::IsTileSupported
bool IsTileSupported(const TensorInfo &input0, const TensorInfo &output, const TileDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1664
armnn::TransposeDescriptor
A TransposeDescriptor for the TransposeLayer.
Definition: Descriptors.hpp:1490
armnn::NeonLayerSupport::IsQuantizeSupported
bool IsQuantizeSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1474
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::SliceDescriptor
A SliceDescriptor for the SliceLayer.
Definition: Descriptors.hpp:1228
IBackendInternal.hpp
armnn::NeonLayerSupport::IsOutputSupported
bool IsOutputSupported(const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1389
armnn::NeonLayerSupport::IsMinimumSupported
bool IsMinimumSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1339
armnn::ReshapeDescriptor
A ReshapeDescriptor for the ReshapeLayer.
Definition: Descriptors.hpp:1023
armnn::NeonLayerSupport::IsPadSupported
bool IsPadSupported(const TensorInfo &input, const TensorInfo &output, const PadDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1395
armnn::NeonLayerSupport::IsDepthToSpaceSupported
bool IsDepthToSpaceSupported(const TensorInfo &input, const TensorInfo &output, const DepthToSpaceDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1041
armnn::NeonLayerSupport::IsTransposeSupported
bool IsTransposeSupported(const TensorInfo &input, const TensorInfo &output, const TransposeDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1692
armnn::PermuteDescriptor
A PermuteDescriptor for the PermuteLayer.
Definition: Descriptors.hpp:149
armnn::BatchMatMulDescriptor
A BatchMatMulDescriptor for the BatchMatMul operator.
Definition: Descriptors.hpp:1584
armnn::SpaceToBatchNdDescriptor
A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer.
Definition: Descriptors.hpp:1043
armnn::NeonLayerSupport::IsQuantizedLstmSupported
bool IsQuantizedLstmSupported(const TensorInfo &input, const TensorInfo &cellStateIn, const TensorInfo &outputStateIn, const TensorInfo &cellStateOut, const TensorInfo &outputStateOut, const QuantizedLstmInputParamsInfo &paramsInfo, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1484
armnn::Convolution3dDescriptor
A Convolution3dDescriptor for the Convolution3dLayer.
Definition: Descriptors.hpp:588
armnn::BaseDescriptor
Base class for all descriptors.
Definition: Descriptors.hpp:22
armnn::NeonLayerSupport::IsConcatSupported
bool IsConcatSupported(const std::vector< const TensorInfo * > inputs, const TensorInfo &output, const OriginsDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:908
armnn::NeonLayerSupport::IsReverseV2Supported
bool IsReverseV2Supported(const TensorInfo &input, const TensorInfo &axis, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported) const
Definition: NeonLayerSupport.cpp:1538
armnn::NeonLayerSupport::IsFullyConnectedSupported
bool IsFullyConnectedSupported(const TensorInfo &input, const TensorInfo &output, const TensorInfo &weights, const TensorInfo &biases, const FullyConnectedDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1176
armnn::NeonLayerSupport::IsElementwiseUnarySupported
bool IsElementwiseUnarySupported(const TensorInfo &input, const TensorInfo &output, const ElementwiseUnaryDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1097
armnn::NeonLayerSupport::IsDepthwiseConvolutionSupported
bool IsDepthwiseConvolutionSupported(const TensorInfo &input, const TensorInfo &output, const DepthwiseConvolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1053
armnn::NeonLayerSupport::IsFillSupported
bool IsFillSupported(const TensorInfo &input, const TensorInfo &output, const FillDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1149
armnn::NeonLayerSupport::~NeonLayerSupport
~NeonLayerSupport()
Definition: NeonLayerSupport.hpp:20
armnn::BatchToSpaceNdDescriptor
A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer.
Definition: Descriptors.hpp:875
armnn::Convolution2dDescriptor
A Convolution2dDescriptor for the Convolution2dLayer.
Definition: Descriptors.hpp:534
armnn::ComparisonDescriptor
A ComparisonDescriptor for the ComparisonLayer.
Definition: Descriptors.hpp:89
armnn::FillDescriptor
A FillDescriptor for the FillLayer.
Definition: Descriptors.hpp:925
armnn::NeonLayerSupport::IsChannelShuffleSupported
bool IsChannelShuffleSupported(const TensorInfo &input, const TensorInfo &output, const ChannelShuffleDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:881
armnn::LstmDescriptor
An LstmDescriptor for the LstmLayer.
Definition: Descriptors.hpp:1102
armnn::StridedSliceDescriptor
A StridedSliceDescriptor for the StridedSliceLayer.
Definition: Descriptors.hpp:1303
armnn::NeonLayerSupport::IsL2NormalizationSupported
bool IsL2NormalizationSupported(const TensorInfo &input, const TensorInfo &output, const L2NormalizationDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1250
armnn::NeonLayerSupport::IsArgMinMaxSupported
bool IsArgMinMaxSupported(const TensorInfo &input, const TensorInfo &output, const ArgMinMaxDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:796
armnn::NeonLayerSupport::IsInstanceNormalizationSupported
bool IsInstanceNormalizationSupported(const TensorInfo &input, const TensorInfo &output, const InstanceNormalizationDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1238
armnn::LogicalBinaryDescriptor
A LogicalBinaryDescriptor for the LogicalBinaryLayer.
Definition: Descriptors.hpp:1518
armnn::NeonLayerSupport::IsTransposeConvolution2dSupported
bool IsTransposeConvolution2dSupported(const TensorInfo &input, const TensorInfo &output, const TransposeConvolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1676
armnn::NeonLayerSupport::IsCastSupported
bool IsCastSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:871
armnn::NeonLayerSupport::IsFloorSupported
bool IsFloorSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1161
armnn::NeonLayerSupport::IsConvolution3dSupported
bool IsConvolution3dSupported(const TensorInfo &input, const TensorInfo &output, const Convolution3dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1008
armnn::NeonLayerSupport::IsInputSupported
bool IsInputSupported(const TensorInfo &input, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1232
armnn::NeonLayerSupport::IsMeanSupported
bool IsMeanSupported(const TensorInfo &input, const TensorInfo &output, const MeanDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1327
armnn::NeonLayerSupport::IsGatherSupported
bool IsGatherSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, const GatherDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported) const
Definition: NeonLayerSupport.cpp:1206
armnn::OriginsDescriptor
An OriginsDescriptor for the ConcatLayer.
Definition: Descriptors.hpp:201
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
armnn::ElementwiseUnaryDescriptor
A ElementwiseUnaryDescriptor for the ElementwiseUnaryLayer.
Definition: Descriptors.hpp:129
armnn::TransposeConvolution2dDescriptor
A TransposeConvolution2dDescriptor for the TransposeConvolution2dLayer.
Definition: Descriptors.hpp:1440
armnn::NeonLayerSupport::NeonLayerSupport
NeonLayerSupport()
Definition: NeonLayerSupport.cpp:179
armnn::LstmInputParamsInfo
Definition: LstmParams.hpp:63
armnn::NeonLayerSupport::IsConvertFp16ToFp32Supported
bool IsConvertFp16ToFp32Supported(const TensorInfo &input, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:955
armnn::NeonLayerSupport::IsDivisionSupported
bool IsDivisionSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1364
armnn::NeonLayerSupport::IsDilatedDepthwiseConvolutionSupported
bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo &input, const TensorInfo &output, const DepthwiseConvolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string & > reason=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1080
armnn::NeonLayerSupport::IsMaximumSupported
bool IsMaximumSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1315
armnn::NeonLayerSupport::IsComparisonSupported
bool IsComparisonSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, const ComparisonDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:893
armnn::Pooling2dDescriptor
A Pooling2dDescriptor for the Pooling2dLayer.
Definition: Descriptors.hpp:371
armnn::DepthwiseConvolution2dDescriptor
A DepthwiseConvolution2dDescriptor for the DepthwiseConvolution2dLayer.
Definition: Descriptors.hpp:659
armnn::NeonLayerSupport::IsUnidirectionalSequenceLstmSupported
bool IsUnidirectionalSequenceLstmSupported(const TensorInfo &input, const TensorInfo &outputStateIn, const TensorInfo &cellStateIn, const TensorInfo &outputStateOut, const TensorInfo &cellStateOut, const TensorInfo &output, const UnidirectionalSequenceLstmDescriptor &descriptor, const LstmInputParamsInfo &paramsInfo, Optional< std::string & > reasonIfUnsupported) const
Definition: NeonLayerSupport.cpp:1700
armnn::ReduceDescriptor
A ReduceDescriptor for the REDUCE operators.
Definition: Descriptors.hpp:1538
armnn::NeonLayerSupport::IsQLstmSupported
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
Definition: NeonLayerSupport.cpp:1439
armnn::NeonLayerSupport::IsActivationSupported
bool IsActivationSupported(const TensorInfo &input, const TensorInfo &output, const ActivationDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:770
armnn::NeonLayerSupport::IsSplitterSupported
bool IsSplitterSupported(const TensorInfo &input, const std::vector< std::reference_wrapper< TensorInfo >> &outputs, const ViewsDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1594
armnn::LayerType
LayerType
When adding a new layer, adapt also the LastLayer enum value in the enum class LayerType below.
Definition: Types.hpp:491
armnn::NeonLayerSupport::IsLogSoftmaxSupported
bool IsLogSoftmaxSupported(const TensorInfo &input, const TensorInfo &output, const LogSoftmaxDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1283
armnn::MeanDescriptor
A MeanDescriptor for the MeanLayer.
Definition: Descriptors.hpp:1172
armnn::TileDescriptor
Definition: Descriptors.hpp:1640
armnn::NeonLayerSupport::IsStridedSliceSupported
bool IsStridedSliceSupported(const TensorInfo &input, const TensorInfo &output, const StridedSliceDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1639
armnn::SoftmaxDescriptor
A SoftmaxDescriptor for the SoftmaxLayer.
Definition: Descriptors.hpp:177
armnn::NeonLayerSupport::IsLstmSupported
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
Definition: NeonLayerSupport.cpp:1291
armnn::IBackendInternal::IBackendSpecificModelContextPtr
std::shared_ptr< IBackendModelContext > IBackendSpecificModelContextPtr
Definition: IBackendInternal.hpp:96
armnn::SpaceToDepthDescriptor
A SpaceToDepthDescriptor for the SpaceToDepthLayer.
Definition: Descriptors.hpp:1075
armnn::NeonLayerSupport::IsPooling2dSupported
bool IsPooling2dSupported(const TensorInfo &input, const TensorInfo &output, const Pooling2dDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:1415
armnn::LayerSupportBase
Definition: LayerSupportBase.hpp:13
armnn::NeonLayerSupport::IsConvolution2dSupported
bool IsConvolution2dSupported(const TensorInfo &input, const TensorInfo &output, const Convolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
Definition: NeonLayerSupport.cpp:975