blob: e7a6748f0add8b1cc1d10cfdf3ff598c7b71027a [file] [log] [blame]
telsoa014fcda012018-03-09 14:13:49 +00001//
Teresa Charlin52664732020-06-29 16:27:03 +01002// Copyright © 2017 Arm Ltd and Contributors. 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
Sadik Armagan045f6be2020-09-10 13:37:32 +01007#include <armnn/backends/IBackendInternal.hpp>
8
Matteo Martincigh992d6dc2019-01-10 17:34:20 +00009#include <backendsCommon/LayerSupportBase.hpp>
telsoa014fcda012018-03-09 14:13:49 +000010
11namespace armnn
12{
arovir014424b0a2018-10-04 10:46:04 +010013
Matteo Martincigh992d6dc2019-01-10 17:34:20 +000014class ClLayerSupport : public LayerSupportBase
arovir014424b0a2018-10-04 10:46:04 +010015{
arovir017c22c702018-10-09 11:16:46 +010016public:
Sadik Armagan045f6be2020-09-10 13:37:32 +010017 explicit ClLayerSupport(const IBackendInternal::IBackendSpecificModelContextPtr& modelContextPtr);
18 ClLayerSupport();
19 ~ClLayerSupport() {}
20
arovir017c22c702018-10-09 11:16:46 +010021 bool IsActivationSupported(const TensorInfo& input,
22 const TensorInfo& output,
23 const ActivationDescriptor& descriptor,
24 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
25
26 bool IsAdditionSupported(const TensorInfo& input0,
27 const TensorInfo& input1,
28 const TensorInfo& output,
29 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
30
James Conroy2dc05722019-09-19 17:00:31 +010031 bool IsArgMinMaxSupported(const TensorInfo& input,
32 const TensorInfo& output,
33 const ArgMinMaxDescriptor& descriptor,
34 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
35
arovir017c22c702018-10-09 11:16:46 +010036 bool IsBatchNormalizationSupported(const TensorInfo& input,
37 const TensorInfo& output,
38 const TensorInfo& mean,
39 const TensorInfo& var,
40 const TensorInfo& beta,
41 const TensorInfo& gamma,
42 const BatchNormalizationDescriptor& descriptor,
43 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
44
Mike Kelly831faed2018-11-28 11:52:08 +000045 bool IsBatchToSpaceNdSupported(const TensorInfo& input,
46 const TensorInfo& output,
47 const BatchToSpaceNdDescriptor& descriptor,
48 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
49
Sadik Armaganf40d6d42021-04-22 09:12:11 +010050 bool IsCastSupported(const TensorInfo& input,
51 const TensorInfo& output,
52 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
53
Teresa Charlin1222dbd2021-09-02 13:58:52 +010054 bool IsChannelShuffleSupported(const TensorInfo& input,
55 const TensorInfo& output,
56 const ChannelShuffleDescriptor& descriptor,
57 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
58
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +010059 bool IsComparisonSupported(const TensorInfo& input0,
60 const TensorInfo& input1,
61 const TensorInfo& ouput,
62 const ComparisonDescriptor& descriptor,
63 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
64
Jim Flynn906f9462019-05-10 13:55:21 +010065 bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,
66 const TensorInfo& output,
Jim Flynne242f2d2019-05-22 14:24:13 +010067 const ConcatDescriptor& descriptor,
Jim Flynn906f9462019-05-10 13:55:21 +010068 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
69
arovir017c22c702018-10-09 11:16:46 +010070 bool IsConstantSupported(const TensorInfo& output,
71 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
72
73 bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
74 const TensorInfo& output,
75 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
76
77 bool IsConvertFp32ToFp16Supported(const TensorInfo& input,
78 const TensorInfo& output,
79 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
80
81 bool IsConvolution2dSupported(const TensorInfo& input,
82 const TensorInfo& output,
83 const Convolution2dDescriptor& descriptor,
84 const TensorInfo& weights,
85 const Optional<TensorInfo>& biases,
86 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
87
Jim Flynn983daec2019-05-29 16:20:16 +010088 bool IsDequantizeSupported(const TensorInfo& input,
89 const TensorInfo& output,
90 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
91
Aron Virginas-Tarb2801962019-09-30 11:24:53 +010092 bool IsDepthToSpaceSupported(const TensorInfo& input,
93 const TensorInfo& output,
94 const DepthToSpaceDescriptor& descriptor,
95 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
96
arovir017c22c702018-10-09 11:16:46 +010097 bool IsDepthwiseConvolutionSupported(const TensorInfo& input,
98 const TensorInfo& output,
99 const DepthwiseConvolution2dDescriptor& descriptor,
100 const TensorInfo& weights,
101 const Optional<TensorInfo>& biases,
102 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
103
Pablo Tellof0bd6832019-04-26 17:58:13 +0100104 bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo& input,
105 const TensorInfo& output,
106 const DepthwiseConvolution2dDescriptor& descriptor,
107 const TensorInfo& weights,
108 const Optional<TensorInfo>& biases,
109 Optional<std::string&> reason = EmptyOptional()) const override;
110
arovir017c22c702018-10-09 11:16:46 +0100111 bool IsDivisionSupported(const TensorInfo& input0,
112 const TensorInfo& input1,
113 const TensorInfo& output,
114 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
115
josh minor4a3c6102020-01-06 16:40:46 -0600116 bool IsElementwiseUnarySupported(const TensorInfo& input,
117 const TensorInfo& ouput,
118 const ElementwiseUnaryDescriptor& descriptor,
119 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
120
Sadik Armagan66aecb02020-06-24 11:42:20 +0100121 bool IsFillSupported(const TensorInfo& input,
122 const TensorInfo& output,
123 const FillDescriptor& descriptor,
124 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
125
arovir017c22c702018-10-09 11:16:46 +0100126 bool IsFloorSupported(const TensorInfo& input,
127 const TensorInfo& output,
128 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
129
130 bool IsFullyConnectedSupported(const TensorInfo& input,
131 const TensorInfo& output,
132 const TensorInfo& weights,
133 const TensorInfo& biases,
134 const FullyConnectedDescriptor& descriptor,
135 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
136
Teresa Charlin9ad2e5b2020-04-10 22:34:48 +0100137 bool IsGatherSupported(const TensorInfo& input0,
138 const TensorInfo& input1,
139 const TensorInfo& output,
Teresa Charlin52664732020-06-29 16:27:03 +0100140 const GatherDescriptor& descriptor,
Teresa Charlin9ad2e5b2020-04-10 22:34:48 +0100141 Optional<std::string&> reasonIfUnsupported) const override;
142
arovir017c22c702018-10-09 11:16:46 +0100143 bool IsInputSupported(const TensorInfo& input,
144 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
145
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100146 bool IsInstanceNormalizationSupported(const TensorInfo& input,
147 const TensorInfo& output,
148 const InstanceNormalizationDescriptor& descriptor,
149 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
150
arovir017c22c702018-10-09 11:16:46 +0100151 bool IsL2NormalizationSupported(const TensorInfo& input,
152 const TensorInfo& output,
153 const L2NormalizationDescriptor& descriptor,
154 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
155
James Conroyfe3ec942020-11-18 14:20:53 +0000156 bool IsLogicalBinarySupported(const TensorInfo& input0,
157 const TensorInfo& input1,
158 const TensorInfo& output,
159 const LogicalBinaryDescriptor& descriptor,
160 Optional<std::string&> reasonIfUnsupported) const override;
161
Teresa Charlin8398edc2020-07-20 14:23:02 +0100162 bool IsLogSoftmaxSupported(const TensorInfo& input,
163 const TensorInfo& output,
164 const LogSoftmaxDescriptor& descriptor,
165 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
166
arovir017c22c702018-10-09 11:16:46 +0100167 bool IsLstmSupported(const TensorInfo& input,
168 const TensorInfo& outputStateIn,
169 const TensorInfo& cellStateIn,
170 const TensorInfo& scratchBuffer,
171 const TensorInfo& outputStateOut,
172 const TensorInfo& cellStateOut,
173 const TensorInfo& output,
174 const LstmDescriptor& descriptor,
Jan Eilersd01a83c2019-07-03 18:20:40 +0100175 const LstmInputParamsInfo& paramsInfo,
176 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
arovir017c22c702018-10-09 11:16:46 +0100177
keidav01a959ee52018-12-19 10:04:58 +0000178 bool IsMaximumSupported(const TensorInfo& input0,
179 const TensorInfo& input1,
180 const TensorInfo& output,
181 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
182
arovir017c22c702018-10-09 11:16:46 +0100183 bool IsMeanSupported(const TensorInfo& input,
184 const TensorInfo& output,
185 const MeanDescriptor& descriptor,
186 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
187
saoste019292aa32019-01-08 13:55:59 +0000188 bool IsMinimumSupported(const TensorInfo& input0,
189 const TensorInfo& input1,
190 const TensorInfo& output,
191 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
192
arovir017c22c702018-10-09 11:16:46 +0100193 bool IsMultiplicationSupported(const TensorInfo& input0,
194 const TensorInfo& input1,
195 const TensorInfo& output,
196 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
197
198 bool IsNormalizationSupported(const TensorInfo& input,
199 const TensorInfo& output,
200 const NormalizationDescriptor& descriptor,
201 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
202
203 bool IsOutputSupported(const TensorInfo& output,
204 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
205
206 bool IsPadSupported(const TensorInfo& input,
207 const TensorInfo& output,
208 const PadDescriptor& descriptor,
209 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
210
211 bool IsPermuteSupported(const TensorInfo& input,
212 const TensorInfo& output,
213 const PermuteDescriptor& descriptor,
214 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
215
216 bool IsPooling2dSupported(const TensorInfo& input,
217 const TensorInfo& output,
218 const Pooling2dDescriptor& descriptor,
219 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
220
Nikhil Raj91e4c6d2019-07-05 12:22:58 +0100221 bool IsPreluSupported(const TensorInfo& input,
222 const TensorInfo& alpha,
223 const TensorInfo& output,
224 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
225
Ryan OShea2323af42020-05-13 16:36:19 +0100226 bool IsQLstmSupported(const TensorInfo& input,
227 const TensorInfo& previousOutputIn,
228 const TensorInfo& previousCellStateIn,
229 const TensorInfo& outputStateOut,
230 const TensorInfo& cellStateOut,
231 const TensorInfo& output,
232 const QLstmDescriptor& descriptor,
233 const LstmInputParamsInfo& paramsInfo,
234 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
235
Ferran Balaguer737d9ff2019-08-01 09:58:08 +0100236 bool IsQuantizedLstmSupported(const TensorInfo& input,
237 const TensorInfo& previousCellStateIn,
238 const TensorInfo& previousOutputIn,
239 const TensorInfo& cellStateOut,
240 const TensorInfo& output,
241 const QuantizedLstmInputParamsInfo& paramsInfo,
242 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
243
Sadik Armagan20ec2492019-05-31 09:09:44 +0100244 bool IsQuantizeSupported(const TensorInfo& input,
245 const TensorInfo& output,
246 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
247
Sadik Armagana2747482021-02-09 10:28:54 +0000248 bool IsReduceSupported(const TensorInfo& input,
249 const TensorInfo& output,
250 const ReduceDescriptor& descriptor,
251 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
252
arovir017c22c702018-10-09 11:16:46 +0100253 bool IsReshapeSupported(const TensorInfo& input,
Kevin Maya023c402019-12-12 17:28:05 +0000254 const TensorInfo& output,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000255 const ReshapeDescriptor& descriptor,
arovir017c22c702018-10-09 11:16:46 +0100256 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
257
Aron Virginas-Tar169d2f12019-07-01 19:01:44 +0100258 bool IsResizeSupported(const TensorInfo& input,
259 const TensorInfo& output,
260 const ResizeDescriptor& descriptor,
261 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
262
Aron Virginas-Tar94c4fef2019-11-25 15:37:08 +0000263 bool IsSliceSupported(const TensorInfo& input,
264 const TensorInfo& output,
265 const SliceDescriptor& descriptor,
266 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
267
arovir017c22c702018-10-09 11:16:46 +0100268 bool IsSoftmaxSupported(const TensorInfo& input,
269 const TensorInfo& output,
270 const SoftmaxDescriptor& descriptor,
271 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
272
Sadik Armaganf4464322018-12-20 16:19:12 +0000273 bool IsSpaceToBatchNdSupported(const TensorInfo& input,
274 const TensorInfo& output,
275 const SpaceToBatchNdDescriptor& descriptor,
276 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
277
James Conroyd2aa85e2019-07-01 17:12:40 +0100278 bool IsSpaceToDepthSupported(const TensorInfo& input,
279 const TensorInfo& output,
280 const SpaceToDepthDescriptor& descriptor,
281 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
282
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100283 bool IsSplitterSupported(const TensorInfo& input,
284 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
285 const ViewsDescriptor& descriptor,
286 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
287
Matthew Jacksond5166102019-07-31 14:06:28 +0100288 bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
289 const TensorInfo& output,
290 const StackDescriptor& descriptor,
291 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
292
keidav01d74dc912018-12-10 18:16:07 +0000293 bool IsStridedSliceSupported(const TensorInfo& input,
294 const TensorInfo& output,
295 const StridedSliceDescriptor& descriptor,
296 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
297
arovir017c22c702018-10-09 11:16:46 +0100298 bool IsSubtractionSupported(const TensorInfo& input0,
299 const TensorInfo& input1,
300 const TensorInfo& output,
301 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Aron Virginas-Tar7a3e2fe2019-06-27 18:54:47 +0100302
303 bool IsTransposeConvolution2dSupported(const TensorInfo& input,
304 const TensorInfo& output,
305 const TransposeConvolution2dDescriptor& descriptor,
306 const TensorInfo& weights,
307 const Optional<TensorInfo>& biases,
308 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000309
310 bool IsTransposeSupported(const TensorInfo& input,
311 const TensorInfo& output,
312 const TransposeDescriptor& descriptor,
313 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
314
Sadik Armagan045f6be2020-09-10 13:37:32 +0100315private:
316 const IBackendInternal::IBackendSpecificModelContextPtr m_ModelContextPtr;
317
arovir014424b0a2018-10-04 10:46:04 +0100318};
319
Aron Virginas-Tarbcf9f162018-10-15 11:47:37 +0100320} // namespace armnn