blob: 67fd230ea2a9dc3994ea9ac9c38bf0fecc3b1320 [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{
arovir014424b0a2018-10-04 10:46:04 +010011
Matteo Martincigh992d6dc2019-01-10 17:34:20 +000012class ClLayerSupport : public LayerSupportBase
arovir014424b0a2018-10-04 10:46:04 +010013{
arovir017c22c702018-10-09 11:16:46 +010014public:
josh minor4a3c6102020-01-06 16:40:46 -060015 ARMNN_DEPRECATED_MSG("Use IsElementwiseUnarySupported instead")
Aron Virginas-Tar82046942019-09-09 15:18:29 +010016 bool IsAbsSupported(const TensorInfo& input,
17 const TensorInfo& output,
18 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
19
arovir017c22c702018-10-09 11:16:46 +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 Conroy2dc05722019-09-19 17:00:31 +010030 bool IsArgMinMaxSupported(const TensorInfo& input,
31 const TensorInfo& output,
32 const ArgMinMaxDescriptor& descriptor,
33 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
34
arovir017c22c702018-10-09 11:16:46 +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 Kelly831faed2018-11-28 11:52:08 +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& ouput,
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
arovir017c22c702018-10-09 11:16:46 +010060 bool IsConstantSupported(const TensorInfo& output,
61 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
62
63 bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
64 const TensorInfo& output,
65 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
66
67 bool IsConvertFp32ToFp16Supported(const TensorInfo& input,
68 const TensorInfo& output,
69 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
70
71 bool IsConvolution2dSupported(const TensorInfo& input,
72 const TensorInfo& output,
73 const Convolution2dDescriptor& descriptor,
74 const TensorInfo& weights,
75 const Optional<TensorInfo>& biases,
76 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
77
Jim Flynn983daec2019-05-29 16:20:16 +010078 bool IsDequantizeSupported(const TensorInfo& input,
79 const TensorInfo& output,
80 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
81
Aron Virginas-Tarb2801962019-09-30 11:24:53 +010082 bool IsDepthToSpaceSupported(const TensorInfo& input,
83 const TensorInfo& output,
84 const DepthToSpaceDescriptor& descriptor,
85 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
86
arovir017c22c702018-10-09 11:16:46 +010087 bool IsDepthwiseConvolutionSupported(const TensorInfo& input,
88 const TensorInfo& output,
89 const DepthwiseConvolution2dDescriptor& descriptor,
90 const TensorInfo& weights,
91 const Optional<TensorInfo>& biases,
92 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
93
Pablo Tellof0bd6832019-04-26 17:58:13 +010094 bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo& input,
95 const TensorInfo& output,
96 const DepthwiseConvolution2dDescriptor& descriptor,
97 const TensorInfo& weights,
98 const Optional<TensorInfo>& biases,
99 Optional<std::string&> reason = EmptyOptional()) const override;
100
arovir017c22c702018-10-09 11:16:46 +0100101 bool IsDivisionSupported(const TensorInfo& input0,
102 const TensorInfo& input1,
103 const TensorInfo& output,
104 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
105
josh minor4a3c6102020-01-06 16:40:46 -0600106 bool IsElementwiseUnarySupported(const TensorInfo& input,
107 const TensorInfo& ouput,
108 const ElementwiseUnaryDescriptor& descriptor,
109 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
110
arovir017c22c702018-10-09 11:16:46 +0100111 bool IsFloorSupported(const TensorInfo& input,
112 const TensorInfo& output,
113 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
114
115 bool IsFullyConnectedSupported(const TensorInfo& input,
116 const TensorInfo& output,
117 const TensorInfo& weights,
118 const TensorInfo& biases,
119 const FullyConnectedDescriptor& descriptor,
120 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
121
Teresa Charlin9ad2e5b2020-04-10 22:34:48 +0100122 bool IsGatherSupported(const TensorInfo& input0,
123 const TensorInfo& input1,
124 const TensorInfo& output,
125 Optional<std::string&> reasonIfUnsupported) const override;
126
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100127 ARMNN_DEPRECATED_MSG("Use IsComparisonSupported instead")
Nattapat Chaimanowongc6a41ff2019-01-29 09:56:02 +0000128 bool IsGreaterSupported(const TensorInfo& input0,
129 const TensorInfo& input1,
130 const TensorInfo& ouput,
131 Optional<std::string&> reasonIfUnsupported) const override;
132
arovir017c22c702018-10-09 11:16:46 +0100133 bool IsInputSupported(const TensorInfo& input,
134 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
135
Aron Virginas-Tar8168f402019-10-04 13:10:16 +0100136 bool IsInstanceNormalizationSupported(const TensorInfo& input,
137 const TensorInfo& output,
138 const InstanceNormalizationDescriptor& descriptor,
139 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
140
arovir017c22c702018-10-09 11:16:46 +0100141 bool IsL2NormalizationSupported(const TensorInfo& input,
142 const TensorInfo& output,
143 const L2NormalizationDescriptor& descriptor,
144 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
145
146 bool IsLstmSupported(const TensorInfo& input,
147 const TensorInfo& outputStateIn,
148 const TensorInfo& cellStateIn,
149 const TensorInfo& scratchBuffer,
150 const TensorInfo& outputStateOut,
151 const TensorInfo& cellStateOut,
152 const TensorInfo& output,
153 const LstmDescriptor& descriptor,
Jan Eilersd01a83c2019-07-03 18:20:40 +0100154 const LstmInputParamsInfo& paramsInfo,
155 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
arovir017c22c702018-10-09 11:16:46 +0100156
keidav01a959ee52018-12-19 10:04:58 +0000157 bool IsMaximumSupported(const TensorInfo& input0,
158 const TensorInfo& input1,
159 const TensorInfo& output,
160 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
161
arovir017c22c702018-10-09 11:16:46 +0100162 bool IsMeanSupported(const TensorInfo& input,
163 const TensorInfo& output,
164 const MeanDescriptor& descriptor,
165 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
166
Jim Flynn906f9462019-05-10 13:55:21 +0100167 ARMNN_DEPRECATED_MSG("Use IsConcatSupported instead")
arovir017c22c702018-10-09 11:16:46 +0100168 bool IsMergerSupported(const std::vector<const TensorInfo*> inputs,
Nikhil Raj8599a412018-11-19 14:51:07 +0000169 const TensorInfo& output,
Jim Flynne242f2d2019-05-22 14:24:13 +0100170 const MergerDescriptor& descriptor,
arovir017c22c702018-10-09 11:16:46 +0100171 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
172
saoste019292aa32019-01-08 13:55:59 +0000173 bool IsMinimumSupported(const TensorInfo& input0,
174 const TensorInfo& input1,
175 const TensorInfo& output,
176 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
177
arovir017c22c702018-10-09 11:16:46 +0100178 bool IsMultiplicationSupported(const TensorInfo& input0,
179 const TensorInfo& input1,
180 const TensorInfo& output,
181 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
182
183 bool IsNormalizationSupported(const TensorInfo& input,
184 const TensorInfo& output,
185 const NormalizationDescriptor& descriptor,
186 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
187
188 bool IsOutputSupported(const TensorInfo& output,
189 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
190
191 bool IsPadSupported(const TensorInfo& input,
192 const TensorInfo& output,
193 const PadDescriptor& descriptor,
194 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
195
196 bool IsPermuteSupported(const TensorInfo& input,
197 const TensorInfo& output,
198 const PermuteDescriptor& descriptor,
199 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
200
201 bool IsPooling2dSupported(const TensorInfo& input,
202 const TensorInfo& output,
203 const Pooling2dDescriptor& descriptor,
204 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
205
Nikhil Raj91e4c6d2019-07-05 12:22:58 +0100206 bool IsPreluSupported(const TensorInfo& input,
207 const TensorInfo& alpha,
208 const TensorInfo& output,
209 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
210
Ryan OShea2323af42020-05-13 16:36:19 +0100211 bool IsQLstmSupported(const TensorInfo& input,
212 const TensorInfo& previousOutputIn,
213 const TensorInfo& previousCellStateIn,
214 const TensorInfo& outputStateOut,
215 const TensorInfo& cellStateOut,
216 const TensorInfo& output,
217 const QLstmDescriptor& descriptor,
218 const LstmInputParamsInfo& paramsInfo,
219 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
220
Ferran Balaguer737d9ff2019-08-01 09:58:08 +0100221 bool IsQuantizedLstmSupported(const TensorInfo& input,
222 const TensorInfo& previousCellStateIn,
223 const TensorInfo& previousOutputIn,
224 const TensorInfo& cellStateOut,
225 const TensorInfo& output,
226 const QuantizedLstmInputParamsInfo& paramsInfo,
227 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
228
Sadik Armagan20ec2492019-05-31 09:09:44 +0100229 bool IsQuantizeSupported(const TensorInfo& input,
230 const TensorInfo& output,
231 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
232
arovir017c22c702018-10-09 11:16:46 +0100233 bool IsReshapeSupported(const TensorInfo& input,
Kevin Maya023c402019-12-12 17:28:05 +0000234 const TensorInfo& output,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000235 const ReshapeDescriptor& descriptor,
arovir017c22c702018-10-09 11:16:46 +0100236 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
237
Aron Virginas-Tar169d2f12019-07-01 19:01:44 +0100238 bool IsResizeSupported(const TensorInfo& input,
239 const TensorInfo& output,
240 const ResizeDescriptor& descriptor,
241 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
242
243 ARMNN_DEPRECATED_MSG("Use IsResizeSupported instead")
arovir017c22c702018-10-09 11:16:46 +0100244 bool IsResizeBilinearSupported(const TensorInfo& input,
Sadik Armaganc625f002018-12-17 11:32:16 +0000245 const TensorInfo& output,
arovir017c22c702018-10-09 11:16:46 +0100246 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
247
josh minor4a3c6102020-01-06 16:40:46 -0600248 ARMNN_DEPRECATED_MSG("Use IsElementwiseUnarySupported instead")
Aron Virginas-Tar1a763dd2019-09-10 12:32:08 +0100249 bool IsRsqrtSupported(const TensorInfo& input,
250 const TensorInfo& output,
251 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
252
Aron Virginas-Tar94c4fef2019-11-25 15:37:08 +0000253 bool IsSliceSupported(const TensorInfo& input,
254 const TensorInfo& output,
255 const SliceDescriptor& descriptor,
256 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
257
arovir017c22c702018-10-09 11:16:46 +0100258 bool IsSoftmaxSupported(const TensorInfo& input,
259 const TensorInfo& output,
260 const SoftmaxDescriptor& descriptor,
261 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
262
Sadik Armaganf4464322018-12-20 16:19:12 +0000263 bool IsSpaceToBatchNdSupported(const TensorInfo& input,
264 const TensorInfo& output,
265 const SpaceToBatchNdDescriptor& descriptor,
266 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
267
James Conroyd2aa85e2019-07-01 17:12:40 +0100268 bool IsSpaceToDepthSupported(const TensorInfo& input,
269 const TensorInfo& output,
270 const SpaceToDepthDescriptor& descriptor,
271 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
272
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100273 ARMNN_DEPRECATED_MSG("Use IsSplitterSupported with outputs instead")
arovir017c22c702018-10-09 11:16:46 +0100274 bool IsSplitterSupported(const TensorInfo& input,
275 const ViewsDescriptor& descriptor,
276 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
277
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100278 bool IsSplitterSupported(const TensorInfo& input,
279 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
280 const ViewsDescriptor& descriptor,
281 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
282
Matthew Jacksond5166102019-07-31 14:06:28 +0100283 bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
284 const TensorInfo& output,
285 const StackDescriptor& descriptor,
286 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
287
keidav01d74dc912018-12-10 18:16:07 +0000288 bool IsStridedSliceSupported(const TensorInfo& input,
289 const TensorInfo& output,
290 const StridedSliceDescriptor& descriptor,
291 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
292
arovir017c22c702018-10-09 11:16:46 +0100293 bool IsSubtractionSupported(const TensorInfo& input0,
294 const TensorInfo& input1,
295 const TensorInfo& output,
296 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Aron Virginas-Tar7a3e2fe2019-06-27 18:54:47 +0100297
298 bool IsTransposeConvolution2dSupported(const TensorInfo& input,
299 const TensorInfo& output,
300 const TransposeConvolution2dDescriptor& descriptor,
301 const TensorInfo& weights,
302 const Optional<TensorInfo>& biases,
303 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000304
305 bool IsTransposeSupported(const TensorInfo& input,
306 const TensorInfo& output,
307 const TransposeDescriptor& descriptor,
308 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
309
arovir014424b0a2018-10-04 10:46:04 +0100310};
311
Aron Virginas-Tarbcf9f162018-10-15 11:47:37 +0100312} // namespace armnn