blob: fda0bc30d3455739fbb5a4cebfd69c6f6a02c34f [file] [log] [blame]
telsoa014fcda012018-03-09 14:13:49 +00001//
2// Copyright © 2017 Arm Ltd. All rights reserved.
David Beckecb56cd2018-09-05 12:52:57 +01003// SPDX-License-Identifier: MIT
telsoa014fcda012018-03-09 14:13:49 +00004//
5#pragma once
6
Matteo Martincigh992d6dc2019-01-10 17:34:20 +00007#include <backendsCommon/LayerSupportBase.hpp>
telsoa014fcda012018-03-09 14:13:49 +00008
9namespace armnn
10{
11
Matteo Martincigh992d6dc2019-01-10 17:34:20 +000012class NeonLayerSupport : public LayerSupportBase
arovir014424b0a2018-10-04 10:46:04 +010013{
arovir017ff76c52018-10-09 09:40:58 +010014public:
josh minor4a3c6102020-01-06 16:40:46 -060015 ARMNN_DEPRECATED_MSG("Use IsElementwiseUnarySupported instead")
Aron Virginas-Tar914e4db2019-09-09 13:36:45 +010016 bool IsAbsSupported(const TensorInfo& input,
17 const TensorInfo& output,
18 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
19
arovir017ff76c52018-10-09 09:40:58 +010020 bool IsActivationSupported(const TensorInfo& input,
21 const TensorInfo& output,
22 const ActivationDescriptor& descriptor,
23 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
24
25 bool IsAdditionSupported(const TensorInfo& input0,
26 const TensorInfo& input1,
27 const TensorInfo& output,
28 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
29
James Conroyd47a0642019-09-17 14:22:06 +010030 bool IsArgMinMaxSupported(const TensorInfo& input,
31 const TensorInfo& output,
32 const ArgMinMaxDescriptor& descriptor,
33 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
34
arovir017ff76c52018-10-09 09:40:58 +010035 bool IsBatchNormalizationSupported(const TensorInfo& input,
36 const TensorInfo& output,
37 const TensorInfo& mean,
38 const TensorInfo& var,
39 const TensorInfo& beta,
40 const TensorInfo& gamma,
41 const BatchNormalizationDescriptor& descriptor,
42 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
43
Mike Kelly56858022020-01-27 12:14:47 +000044 bool IsBatchToSpaceNdSupported(const TensorInfo& input,
45 const TensorInfo& output,
46 const BatchToSpaceNdDescriptor& descriptor,
47 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
48
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +010049 bool IsComparisonSupported(const TensorInfo& input0,
50 const TensorInfo& input1,
51 const TensorInfo& output,
52 const ComparisonDescriptor& descriptor,
53 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
54
Jim Flynn906f9462019-05-10 13:55:21 +010055 bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,
56 const TensorInfo& output,
Jim Flynne242f2d2019-05-22 14:24:13 +010057 const ConcatDescriptor& descriptor,
Jim Flynn906f9462019-05-10 13:55:21 +010058 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
59
arovir017ff76c52018-10-09 09:40:58 +010060 bool IsConstantSupported(const TensorInfo& output,
61 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
62
Narumol Prangnawarat250d3922020-03-30 16:11:04 +010063 bool IsConvertBf16ToFp32Supported(const TensorInfo& input,
64 const TensorInfo& output,
65 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
66
arovir017ff76c52018-10-09 09:40:58 +010067 bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
68 const TensorInfo& output,
69 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
70
Narumol Prangnawarat250d3922020-03-30 16:11:04 +010071 bool IsConvertFp32ToBf16Supported(const TensorInfo& input,
72 const TensorInfo& output,
73 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
74
arovir017ff76c52018-10-09 09:40:58 +010075 bool IsConvertFp32ToFp16Supported(const TensorInfo& input,
76 const TensorInfo& output,
77 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
78
79 bool IsConvolution2dSupported(const TensorInfo& input,
80 const TensorInfo& output,
81 const Convolution2dDescriptor& descriptor,
82 const TensorInfo& weights,
83 const Optional<TensorInfo>& biases,
84 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
85
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +010086 bool IsDepthToSpaceSupported(const TensorInfo& input,
87 const TensorInfo& output,
88 const DepthToSpaceDescriptor& descriptor,
89 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
90
91
arovir017ff76c52018-10-09 09:40:58 +010092 bool IsDepthwiseConvolutionSupported(const TensorInfo& input,
93 const TensorInfo& output,
94 const DepthwiseConvolution2dDescriptor& descriptor,
95 const TensorInfo& weights,
96 const Optional<TensorInfo>& biases,
97 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
98
Narumol Prangnawarat01961a72019-05-30 16:47:12 +010099 bool IsDequantizeSupported(const TensorInfo& input,
100 const TensorInfo& output,
101 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
102
Pablo Tellof0bd6832019-04-26 17:58:13 +0100103 bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo& input,
104 const TensorInfo& output,
105 const DepthwiseConvolution2dDescriptor& descriptor,
106 const TensorInfo& weights,
107 const Optional<TensorInfo>& biases,
108 Optional<std::string&> reason = EmptyOptional()) const override;
109
josh minor4a3c6102020-01-06 16:40:46 -0600110 bool IsElementwiseUnarySupported(const TensorInfo& input,
111 const TensorInfo& output,
112 const ElementwiseUnaryDescriptor& descriptor,
113 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
114
arovir017ff76c52018-10-09 09:40:58 +0100115 bool IsFloorSupported(const TensorInfo& input,
116 const TensorInfo& output,
117 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
118
119 bool IsFullyConnectedSupported(const TensorInfo& input,
120 const TensorInfo& output,
121 const TensorInfo& weights,
122 const TensorInfo& biases,
123 const FullyConnectedDescriptor& descriptor,
124 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
125
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100126 ARMNN_DEPRECATED_MSG("Use IsComparisonSupported instead")
kevmay01eed85922019-01-28 08:37:25 +0000127 bool IsGreaterSupported(const TensorInfo& input0,
128 const TensorInfo& input1,
129 const TensorInfo& output,
130 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
131
arovir017ff76c52018-10-09 09:40:58 +0100132 bool IsInputSupported(const TensorInfo& input,
133 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
134
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100135 bool IsInstanceNormalizationSupported(const TensorInfo& input,
136 const TensorInfo& output,
137 const InstanceNormalizationDescriptor& descriptor,
138 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
139
arovir017ff76c52018-10-09 09:40:58 +0100140 bool IsL2NormalizationSupported(const TensorInfo& input,
141 const TensorInfo& output,
142 const L2NormalizationDescriptor& descriptor,
143 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
144
Jan Eilersad5293a2019-07-08 09:57:55 +0100145 bool IsLstmSupported(const TensorInfo& input,
146 const TensorInfo& outputStateIn,
147 const TensorInfo& cellStateIn,
148 const TensorInfo& scratchBuffer,
149 const TensorInfo& outputStateOut,
150 const TensorInfo& cellStateOut,
151 const TensorInfo& output,
152 const LstmDescriptor& descriptor,
153 const LstmInputParamsInfo& paramsInfo,
154 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
155
Nattapat Chaimanowong4e6597a2018-12-20 14:14:06 +0000156 bool IsMaximumSupported(const TensorInfo& input0,
157 const TensorInfo& input1,
158 const TensorInfo& output,
159 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
160
arovir017ff76c52018-10-09 09:40:58 +0100161 bool IsMeanSupported(const TensorInfo& input,
162 const TensorInfo& output,
163 const MeanDescriptor& descriptor,
164 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
165
Jim Flynn906f9462019-05-10 13:55:21 +0100166 ARMNN_DEPRECATED_MSG("Use IsConcatSupported instead")
arovir017ff76c52018-10-09 09:40:58 +0100167 bool IsMergerSupported(const std::vector<const TensorInfo*> inputs,
Nikhil Raj8599a412018-11-19 14:51:07 +0000168 const TensorInfo& output,
Jim Flynne242f2d2019-05-22 14:24:13 +0100169 const MergerDescriptor& descriptor,
arovir017ff76c52018-10-09 09:40:58 +0100170 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
171
Conor Kennedy54b21692019-01-09 07:57:38 +0000172 bool IsMinimumSupported(const TensorInfo& input0,
173 const TensorInfo& input1,
174 const TensorInfo& output,
175 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
176
arovir017ff76c52018-10-09 09:40:58 +0100177 bool IsMultiplicationSupported(const TensorInfo& input0,
178 const TensorInfo& input1,
179 const TensorInfo& output,
180 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
181
Pablo Telloe61f0712020-01-23 10:37:17 +0000182 bool IsDivisionSupported(const TensorInfo& input0,
183 const TensorInfo& input1,
184 const TensorInfo& output,
185 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
186
arovir017ff76c52018-10-09 09:40:58 +0100187 bool IsNormalizationSupported(const TensorInfo& input,
188 const TensorInfo& output,
189 const NormalizationDescriptor& descriptor,
190 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
191
192 bool IsOutputSupported(const TensorInfo& output,
193 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
194
Éanna Ó Catháin12055742019-01-25 10:01:40 +0000195 bool IsPadSupported(const TensorInfo& input,
196 const TensorInfo& output,
197 const PadDescriptor& descriptor,
198 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
199
arovir017ff76c52018-10-09 09:40:58 +0100200 bool IsPermuteSupported(const TensorInfo& input,
201 const TensorInfo& output,
202 const PermuteDescriptor& descriptor,
203 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
204
205 bool IsPooling2dSupported(const TensorInfo& input,
206 const TensorInfo& output,
207 const Pooling2dDescriptor& descriptor,
208 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
209
Nikhil Raj9b461482019-07-03 15:58:31 +0100210 bool IsPreluSupported(const TensorInfo& input,
211 const TensorInfo& alpha,
212 const TensorInfo& output,
213 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
214
Sadik Armaganfabc2892019-05-31 09:05:11 +0100215 bool IsQuantizeSupported(const TensorInfo& input,
216 const TensorInfo& output,
217 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
218
Francis Murtagh4fc3c482019-08-02 13:20:54 +0100219 bool IsQuantizedLstmSupported(const TensorInfo& input,
220 const TensorInfo& cellStateIn,
221 const TensorInfo& outputStateIn,
222 const TensorInfo& cellStateOut,
223 const TensorInfo& outputStateOut,
224 const QuantizedLstmInputParamsInfo& paramsInfo,
225 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
226
arovir017ff76c52018-10-09 09:40:58 +0100227 bool IsReshapeSupported(const TensorInfo& input,
Kevin Maya023c402019-12-12 17:28:05 +0000228 const TensorInfo& output,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000229 const ReshapeDescriptor& descriptor,
arovir017ff76c52018-10-09 09:40:58 +0100230 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
231
Aron Virginas-Tar169d2f12019-07-01 19:01:44 +0100232 bool IsResizeSupported(const TensorInfo& input,
233 const TensorInfo& output,
234 const ResizeDescriptor& descriptor,
235 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
236
237 ARMNN_DEPRECATED_MSG("Use IsResizeSupported instead")
Sadik Armaganc625f002018-12-17 11:32:16 +0000238 bool IsResizeBilinearSupported(const TensorInfo& input,
239 const TensorInfo& output,
240 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
241
josh minor4a3c6102020-01-06 16:40:46 -0600242 ARMNN_DEPRECATED_MSG("Use IsElementwiseUnarySupported instead")
Aron Virginas-Tar0dd3b432019-09-10 13:55:09 +0100243 bool IsRsqrtSupported(const TensorInfo& input,
244 const TensorInfo& output,
245 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
246
josh minor036f02d2019-11-15 14:53:22 -0600247 bool IsSliceSupported(const TensorInfo& input,
248 const TensorInfo& output,
249 const SliceDescriptor& descriptor,
250 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
251
arovir017ff76c52018-10-09 09:40:58 +0100252 bool IsSoftmaxSupported(const TensorInfo& input,
253 const TensorInfo& output,
254 const SoftmaxDescriptor& descriptor,
255 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
256
Mike Kelly0be3a882020-01-24 11:27:50 +0000257 bool IsSpaceToBatchNdSupported(const TensorInfo& input,
258 const TensorInfo& output,
259 const SpaceToBatchNdDescriptor& descriptor,
260 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
261
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100262 bool IsSpaceToDepthSupported(const TensorInfo& input,
263 const TensorInfo& output,
264 const SpaceToDepthDescriptor& descriptor,
265 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
266
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100267 ARMNN_DEPRECATED_MSG("Use IsSplitterSupported with outputs instead")
arovir017ff76c52018-10-09 09:40:58 +0100268 bool IsSplitterSupported(const TensorInfo& input,
269 const ViewsDescriptor& descriptor,
270 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
271
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100272 bool IsSplitterSupported(const TensorInfo& input,
273 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
274 const ViewsDescriptor& descriptor,
275 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
276
Matthew Jackson87f65ea2019-08-01 10:01:34 +0100277 bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
278 const TensorInfo& output,
279 const StackDescriptor& descriptor,
280 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
281
FinnWilliamsArm1fa19192019-08-02 17:26:31 +0100282 bool IsStridedSliceSupported(const TensorInfo& input,
283 const TensorInfo& output,
284 const StridedSliceDescriptor& descriptor,
285 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
286
arovir017ff76c52018-10-09 09:40:58 +0100287 bool IsSubtractionSupported(const TensorInfo& input0,
288 const TensorInfo& input1,
289 const TensorInfo& output,
290 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Sadik Armagan581742d2019-08-12 14:11:37 +0100291
292 bool IsTransposeConvolution2dSupported(const TensorInfo& input,
293 const TensorInfo& output,
294 const TransposeConvolution2dDescriptor& descriptor,
295 const TensorInfo& weights,
296 const Optional<TensorInfo>& biases,
297 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
298
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000299 bool IsTransposeSupported(const TensorInfo& input,
300 const TensorInfo& output,
301 const TransposeDescriptor& descriptor,
302 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
303
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100304}; // class NeonLayerSupport
arovir014424b0a2018-10-04 10:46:04 +0100305
arovir017ff76c52018-10-09 09:40:58 +0100306} // namespace armnn