From 549b9600a6eaf0727fa084465a75f173edf8f381 Mon Sep 17 00:00:00 2001 From: Nikhil Raj Date: Tue, 24 May 2022 11:32:07 +0100 Subject: Update 22.05 Doxygen Docs after updates to main Readme Signed-off-by: Nikhil Raj Change-Id: I56711772406a41ff81fa136a5fb6c59c9b9cf504 --- 22.05/_neon_layer_support_8hpp_source.xhtml | 247 ++++++++++++++++++++++++++++ 1 file changed, 247 insertions(+) create mode 100644 22.05/_neon_layer_support_8hpp_source.xhtml (limited to '22.05/_neon_layer_support_8hpp_source.xhtml') diff --git a/22.05/_neon_layer_support_8hpp_source.xhtml b/22.05/_neon_layer_support_8hpp_source.xhtml new file mode 100644 index 0000000000..41b59e429d --- /dev/null +++ b/22.05/_neon_layer_support_8hpp_source.xhtml @@ -0,0 +1,247 @@ + + + + + + + + + + + + + +ArmNN: src/backends/neon/NeonLayerSupport.hpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  22.05 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
NeonLayerSupport.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 #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,
25  const Optional<LstmInputParamsInfo>& lstmParamsInfo,
26  const Optional<QuantizedLstmInputParamsInfo>& quantizedLstmParamsInfo,
28 
29  bool IsActivationSupported(const TensorInfo& input,
30  const TensorInfo& output,
31  const ActivationDescriptor& descriptor,
32  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
33 
34  bool IsAdditionSupported(const TensorInfo& input0,
35  const TensorInfo& input1,
36  const TensorInfo& output,
37  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
38 
39  bool IsArgMinMaxSupported(const TensorInfo& input,
40  const TensorInfo& output,
41  const ArgMinMaxDescriptor& descriptor,
42  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
43 
44  bool IsBatchNormalizationSupported(const TensorInfo& input,
45  const TensorInfo& output,
46  const TensorInfo& mean,
47  const TensorInfo& var,
48  const TensorInfo& beta,
49  const TensorInfo& gamma,
50  const BatchNormalizationDescriptor& descriptor,
51  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
52 
53  bool IsBatchToSpaceNdSupported(const TensorInfo& input,
54  const TensorInfo& output,
55  const BatchToSpaceNdDescriptor& descriptor,
56  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
57 
58  bool IsCastSupported(const TensorInfo& input,
59  const TensorInfo& output,
60  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
61 
62  bool IsChannelShuffleSupported(const TensorInfo& input,
63  const TensorInfo& output,
64  const ChannelShuffleDescriptor& descriptor,
65  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
66 
67  bool IsComparisonSupported(const TensorInfo& input0,
68  const TensorInfo& input1,
69  const TensorInfo& output,
70  const ComparisonDescriptor& descriptor,
71  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
72 
73  bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,
74  const TensorInfo& output,
75  const OriginsDescriptor& descriptor,
76  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
77 
78  bool IsConstantSupported(const TensorInfo& output,
79  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
80 
81  bool IsConvertBf16ToFp32Supported(const TensorInfo& input,
82  const TensorInfo& output,
83  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
84 
85  bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
86  const TensorInfo& output,
87  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
88 
89  bool IsConvertFp32ToBf16Supported(const TensorInfo& input,
90  const TensorInfo& output,
91  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
92 
93  bool IsConvertFp32ToFp16Supported(const TensorInfo& input,
94  const TensorInfo& output,
95  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
96 
97  bool IsConvolution2dSupported(const TensorInfo& input,
98  const TensorInfo& output,
99  const Convolution2dDescriptor& descriptor,
100  const TensorInfo& weights,
102  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
103 
104  bool IsConvolution3dSupported(const TensorInfo& input,
105  const TensorInfo& output,
106  const Convolution3dDescriptor& descriptor,
107  const TensorInfo& weights,
108  const Optional<TensorInfo>& biases,
109  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
110 
111  bool IsDepthToSpaceSupported(const TensorInfo& input,
112  const TensorInfo& output,
113  const DepthToSpaceDescriptor& descriptor,
114  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
115 
116 
118  const TensorInfo& output,
119  const DepthwiseConvolution2dDescriptor& descriptor,
120  const TensorInfo& weights,
121  const Optional<TensorInfo>& biases,
122  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
123 
124  bool IsDequantizeSupported(const TensorInfo& input,
125  const TensorInfo& output,
126  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
127 
129  const TensorInfo& output,
130  const DepthwiseConvolution2dDescriptor& descriptor,
131  const TensorInfo& weights,
132  const Optional<TensorInfo>& biases,
133  Optional<std::string&> reason = EmptyOptional()) const override;
134 
135  bool IsElementwiseUnarySupported(const TensorInfo& input,
136  const TensorInfo& output,
137  const ElementwiseUnaryDescriptor& descriptor,
138  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
139 
140  bool IsFillSupported(const TensorInfo& input,
141  const TensorInfo& output,
142  const FillDescriptor& descriptor,
143  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
144 
145  bool IsFloorSupported(const TensorInfo& input,
146  const TensorInfo& output,
147  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
148 
149  bool IsFullyConnectedSupported(const TensorInfo& input,
150  const TensorInfo& output,
151  const TensorInfo& weights,
152  const TensorInfo& biases,
153  const FullyConnectedDescriptor& descriptor,
154  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
155 
156  bool IsGatherNdSupported(const TensorInfo& input0,
157  const TensorInfo& input1,
158  const TensorInfo& output,
159  Optional<std::string&> reasonIfUnsupported) const;
160 
161  bool IsGatherSupported(const TensorInfo& input0,
162  const TensorInfo& input1,
163  const TensorInfo& output,
164  const GatherDescriptor& descriptor,
165  Optional<std::string&> reasonIfUnsupported) const override;
166 
167  bool IsInputSupported(const TensorInfo& input,
168  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
169 
171  const TensorInfo& output,
172  const InstanceNormalizationDescriptor& descriptor,
173  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
174 
175  bool IsL2NormalizationSupported(const TensorInfo& input,
176  const TensorInfo& output,
177  const L2NormalizationDescriptor& descriptor,
178  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
179 
180  bool IsLogicalBinarySupported(const TensorInfo& input0,
181  const TensorInfo& input1,
182  const TensorInfo& output,
183  const LogicalBinaryDescriptor& descriptor,
184  Optional<std::string&> reasonIfUnsupported) const override;
185 
186  bool IsLogSoftmaxSupported(const TensorInfo& input,
187  const TensorInfo& output,
188  const LogSoftmaxDescriptor& descriptor,
189  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
190 
191  bool IsLstmSupported(const TensorInfo& input,
192  const TensorInfo& outputStateIn,
193  const TensorInfo& cellStateIn,
194  const TensorInfo& scratchBuffer,
195  const TensorInfo& outputStateOut,
196  const TensorInfo& cellStateOut,
197  const TensorInfo& output,
198  const LstmDescriptor& descriptor,
200  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
201 
202  bool IsMaximumSupported(const TensorInfo& input0,
203  const TensorInfo& input1,
204  const TensorInfo& output,
205  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
206 
207  bool IsMeanSupported(const TensorInfo& input,
208  const TensorInfo& output,
209  const MeanDescriptor& descriptor,
210  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
211 
212  bool IsMinimumSupported(const TensorInfo& input0,
213  const TensorInfo& input1,
214  const TensorInfo& output,
215  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
216 
217  bool IsMultiplicationSupported(const TensorInfo& input0,
218  const TensorInfo& input1,
219  const TensorInfo& output,
220  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
221 
222  bool IsDivisionSupported(const TensorInfo& input0,
223  const TensorInfo& input1,
224  const TensorInfo& output,
225  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
226 
227  bool IsNormalizationSupported(const TensorInfo& input,
228  const TensorInfo& output,
229  const NormalizationDescriptor& descriptor,
230  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
231 
232  bool IsOutputSupported(const TensorInfo& output,
233  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
234 
235  bool IsPadSupported(const TensorInfo& input,
236  const TensorInfo& output,
237  const PadDescriptor& descriptor,
238  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
239 
240  bool IsPermuteSupported(const TensorInfo& input,
241  const TensorInfo& output,
242  const PermuteDescriptor& descriptor,
243  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
244 
245  bool IsPooling2dSupported(const TensorInfo& input,
246  const TensorInfo& output,
247  const Pooling2dDescriptor& descriptor,
248  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
249  bool IsPooling3dSupported(const TensorInfo& input,
250  const TensorInfo& output,
251  const Pooling3dDescriptor& descriptor,
252  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
253  bool IsPreluSupported(const TensorInfo& input,
254  const TensorInfo& alpha,
255  const TensorInfo& output,
256  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
257 
258  bool IsQLstmSupported(const TensorInfo& input,
261  const TensorInfo& outputStateOut,
262  const TensorInfo& cellStateOut,
263  const TensorInfo& output,
264  const QLstmDescriptor& descriptor,
265  const LstmInputParamsInfo& paramsInfo,
266  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
267 
268  bool IsQuantizeSupported(const TensorInfo& input,
269  const TensorInfo& output,
270  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
271 
272  bool IsQuantizedLstmSupported(const TensorInfo& input,
273  const TensorInfo& cellStateIn,
274  const TensorInfo& outputStateIn,
275  const TensorInfo& cellStateOut,
276  const TensorInfo& outputStateOut,
277  const QuantizedLstmInputParamsInfo& paramsInfo,
278  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
279 
280  bool IsReduceSupported(const TensorInfo& input,
281  const TensorInfo& output,
282  const ReduceDescriptor& descriptor,
283  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
284 
285  bool IsReshapeSupported(const TensorInfo& input,
286  const TensorInfo& output,
287  const ReshapeDescriptor& descriptor,
288  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
289 
290  bool IsResizeSupported(const TensorInfo& input,
291  const TensorInfo& output,
292  const ResizeDescriptor& descriptor,
293  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
294 
295  bool IsSliceSupported(const TensorInfo& input,
296  const TensorInfo& output,
297  const SliceDescriptor& descriptor,
298  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
299 
300  bool IsSoftmaxSupported(const TensorInfo& input,
301  const TensorInfo& output,
302  const SoftmaxDescriptor& descriptor,
303  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
304 
305  bool IsSpaceToBatchNdSupported(const TensorInfo& input,
306  const TensorInfo& output,
307  const SpaceToBatchNdDescriptor& descriptor,
308  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
309 
310  bool IsSpaceToDepthSupported(const TensorInfo& input,
311  const TensorInfo& output,
312  const SpaceToDepthDescriptor& descriptor,
313  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
314 
315  bool IsSplitterSupported(const TensorInfo& input,
316  const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
317  const ViewsDescriptor& descriptor,
318  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
319 
320  bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
321  const TensorInfo& output,
322  const StackDescriptor& descriptor,
323  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
324 
325  bool IsStridedSliceSupported(const TensorInfo& input,
326  const TensorInfo& output,
327  const StridedSliceDescriptor& descriptor,
328  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
329 
330  bool IsSubtractionSupported(const TensorInfo& input0,
331  const TensorInfo& input1,
332  const TensorInfo& output,
333  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
334 
336  const TensorInfo& output,
337  const TransposeConvolution2dDescriptor& descriptor,
338  const TensorInfo& weights,
339  const Optional<TensorInfo>& biases,
340  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
341 
342  bool IsTransposeSupported(const TensorInfo& input,
343  const TensorInfo& output,
344  const TransposeDescriptor& descriptor,
345  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
346 
348  const TensorInfo& outputStateIn,
349  const TensorInfo& cellStateIn,
350  const TensorInfo& outputStateOut,
351  const TensorInfo& cellStateOut,
352  const TensorInfo& output,
353  const UnidirectionalSequenceLstmDescriptor& descriptor,
354  const LstmInputParamsInfo& paramsInfo,
355  Optional<std::string&> reasonIfUnsupported) const override;
356 
357 private:
359 
360 }; // class NeonLayerSupport
361 
362 } // namespace armnn
+
bool IsConcatSupported(const std::vector< const TensorInfo *> inputs, const TensorInfo &output, const OriginsDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
bool IsSliceSupported(const TensorInfo &input, const TensorInfo &output, const SliceDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
bool IsSoftmaxSupported(const TensorInfo &input, const TensorInfo &output, const SoftmaxDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
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
+
A ViewsDescriptor for the SplitterLayer.
+
bool IsConvertFp32ToFp16Supported(const TensorInfo &input, const TensorInfo &output, 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
+ +
const TensorInfo & output
+
A ReshapeDescriptor for the ReshapeLayer.
+
bool IsGatherSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, const GatherDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported) const override
+
bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo &input, const TensorInfo &output, const DepthwiseConvolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, Optional< std::string &> reason=EmptyOptional()) const override
+ +
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 override
+ +
A ComparisonDescriptor for the ComparisonLayer.
Definition: Descriptors.hpp:89
+
bool IsDepthToSpaceSupported(const TensorInfo &input, const TensorInfo &output, const DepthToSpaceDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
const TensorInfo const TensorInfo const TensorInfo const TensorInfo const TensorInfo & gamma
+
bool IsL2NormalizationSupported(const TensorInfo &input, const TensorInfo &output, const L2NormalizationDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
bool IsCastSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
bool IsStridedSliceSupported(const TensorInfo &input, const TensorInfo &output, const StridedSliceDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
const std::vector< std::reference_wrapper< TensorInfo > > & outputs
+ +
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
+
bool IsConvertFp32ToBf16Supported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
bool IsAdditionSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
A Convolution2dDescriptor for the Convolution2dLayer.
+
bool IsGatherNdSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported) const
+
const TensorInfo const ActivationDescriptor Optional< std::string & > reasonIfUnsupported
+
A LogicalBinaryDescriptor for the LogicalBinaryLayer.
+
bool IsConstantSupported(const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
Copyright (c) 2021 ARM Limited and Contributors.
+
bool IsPadSupported(const TensorInfo &input, const TensorInfo &output, const PadDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
const TensorInfo const ActivationDescriptor & descriptor
+ +
A SpaceToDepthDescriptor for the SpaceToDepthLayer.
+ +
bool IsPooling2dSupported(const TensorInfo &input, const TensorInfo &output, const Pooling2dDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer.
+
bool IsActivationSupported(const TensorInfo &input, const TensorInfo &output, const ActivationDescriptor &descriptor, 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
+
const TensorInfo & outputStateIn
+
const TensorInfo const TensorInfo & previousCellStateIn
+
bool IsComparisonSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, const ComparisonDescriptor &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 IsLogicalBinarySupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, const LogicalBinaryDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported) const override
+
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
+
const TensorInfo const TensorInfo & cellStateIn
+
bool IsFillSupported(const TensorInfo &input, const TensorInfo &output, const FillDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
An LstmDescriptor for the LstmLayer.
+
bool IsInputSupported(const TensorInfo &input, 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
+ +
std::shared_ptr< IBackendModelContext > IBackendSpecificModelContextPtr
+
A L2NormalizationDescriptor for the L2NormalizationLayer.
+
const TensorInfo const TensorInfo const TensorInfo & var
+
bool IsMinimumSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
bool IsPreluSupported(const TensorInfo &input, const TensorInfo &alpha, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
bool IsOutputSupported(const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
An ArgMinMaxDescriptor for ArgMinMaxLayer.
Definition: Descriptors.hpp:67
+
bool IsInstanceNormalizationSupported(const TensorInfo &input, const TensorInfo &output, const InstanceNormalizationDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
An OriginsDescriptor for the ConcatLayer.
+
A ReduceDescriptor for the REDUCE operators.
+
A FullyConnectedDescriptor for the FullyConnectedLayer.
+
const TensorInfo const TensorInfo const TensorInfo const TensorInfo & outputStateOut
+
bool IsNormalizationSupported(const TensorInfo &input, const TensorInfo &output, const NormalizationDescriptor &descriptor, 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
+
const TensorInfo const TensorInfo const TensorInfo const TensorInfo const TensorInfo & cellStateOut
+
A GatherDescriptor for the GatherLayer.
+
bool IsDivisionSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
const TensorInfo const TensorInfo const TensorInfo const TensorInfo & beta
+
bool IsFloorSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
A QLstmDescriptor for the QLstmLayer.
+
bool IsLogSoftmaxSupported(const TensorInfo &input, const TensorInfo &output, const LogSoftmaxDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
An ActivationDescriptor for the ActivationLayer.
Definition: Descriptors.hpp:36
+
bool IsResizeSupported(const TensorInfo &input, const TensorInfo &output, const ResizeDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
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
+
A SliceDescriptor for the SliceLayer.
+
A Convolution3dDescriptor for the Convolution3dLayer.
+
bool IsSplitterSupported(const TensorInfo &input, const std::vector< std::reference_wrapper< TensorInfo >> &outputs, const ViewsDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
const TensorInfo & previousOutputIn
+
bool IsReshapeSupported(const TensorInfo &input, const TensorInfo &output, const ReshapeDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
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 Pooling3dDescriptor for the Pooling3dLayer.
+
A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer.
+
bool IsQuantizeSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
EmptyOptional is used to initialize the Optional class in case we want to have default value for an O...
Definition: Optional.hpp:32
+
bool IsConvertFp16ToFp32Supported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
A ElementwiseUnaryDescriptor for the ElementwiseUnaryLayer.
+
bool IsSubtractionSupported(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
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
+
bool IsSpaceToBatchNdSupported(const TensorInfo &input, const TensorInfo &output, const SpaceToBatchNdDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
const TensorInfo const Convolution2dDescriptor const TensorInfo const Optional< TensorInfo > & biases
+
bool IsPooling3dSupported(const TensorInfo &input, const TensorInfo &output, const Pooling3dDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
A MeanDescriptor for the MeanLayer.
+
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 IsArgMinMaxSupported(const TensorInfo &input, const TensorInfo &output, const ArgMinMaxDescriptor &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
+
A TransposeDescriptor for the TransposeLayer.
+
A StridedSliceDescriptor for the StridedSliceLayer.
+
bool IsTransposeSupported(const TensorInfo &input, const TensorInfo &output, const TransposeDescriptor &descriptor, 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 & input1
+
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 IsMeanSupported(const TensorInfo &input, const TensorInfo &output, const MeanDescriptor &descriptor, 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
+
A Pooling2dDescriptor for the Pooling2dLayer.
+
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 override
+
A NormalizationDescriptor for the NormalizationLayer.
+
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
+
bool IsConvertBf16ToFp32Supported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
const TensorInfo const TensorInfo const TensorInfo & scratchBuffer
+
An InstanceNormalizationDescriptor for InstanceNormalizationLayer.
+
A ChannelShuffleDescriptor for the ChannelShuffle operator.
+
A SoftmaxDescriptor for the SoftmaxLayer.
+
bool IsReduceSupported(const TensorInfo &input, const TensorInfo &output, const ReduceDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
const TensorInfo const Convolution2dDescriptor const TensorInfo & weights
+
bool IsDequantizeSupported(const TensorInfo &input, const TensorInfo &output, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
A DepthwiseConvolution2dDescriptor for the DepthwiseConvolution2dLayer.
+
A FillDescriptor for the FillLayer.
+
A BatchNormalizationDescriptor for the BatchNormalizationLayer.
+
bool IsChannelShuffleSupported(const TensorInfo &input, const TensorInfo &output, const ChannelShuffleDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
const TensorInfo const TensorInfo & mean
+
A PermuteDescriptor for the PermuteLayer.
+
LayerType
When adding a new layer, adapt also the LastLayer enum value in the enum class LayerType below...
Definition: Types.hpp:467
+
bool IsPermuteSupported(const TensorInfo &input, const TensorInfo &output, const PermuteDescriptor &descriptor, Optional< std::string &> reasonIfUnsupported=EmptyOptional()) const override
+
+
+ + + + -- cgit v1.2.1