Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 25334cf3d53fe7fff98776b44a199ca341f62f1a / . / src / backends / reference / RefLayerSupport.hpp
blob: ff3478142fcceec8ee91f1577cf31fe6809c1649 [file] [log] [blame]
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1//
2// Copyright © 2017 Arm Ltd. All rights reserved.
David Beckecb56cd2018-09-05 12:52:57 +0100[diff] [blame]3// SPDX-License-Identifier: MIT
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]4//
5#pragma once
6
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]7#include <backendsCommon/LayerSupportBase.hpp>
arovir01a6824102018-08-28 17:40:45 +0100[diff] [blame]8
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]9namespace armnn
10{
11
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]12class RefLayerSupport : public LayerSupportBase
arovir014424b0a2018-10-04 10:46:04 +0100[diff] [blame]13{
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]14public:
josh minor4a3c6102020-01-06 16:40:46 -0600[diff] [blame]15 ARMNN_DEPRECATED_MSG("Use IsElementwiseUnarySupported instead")
Sadik Armagan9199e582019-09-05 17:35:31 +0100[diff] [blame]16 bool IsAbsSupported(const TensorInfo& input,
17 const TensorInfo& output,
18 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
19
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]20 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
Nikhil Raj68c2c902019-09-19 11:21:11 +0100[diff] [blame]30 bool IsArgMinMaxSupported(const TensorInfo& input,
31 const TensorInfo& output,
32 const ArgMinMaxDescriptor& descriptor,
33 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
34
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]35 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
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000[diff] [blame]44 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 +0100[diff] [blame]49 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 +0100[diff] [blame]55 bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,
56 const TensorInfo& output,
Jim Flynne242f2d2019-05-22 14:24:13 +0100[diff] [blame]57 const ConcatDescriptor& descriptor,
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]58 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
59
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]60 bool IsConstantSupported(const TensorInfo& output,
61 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
62
Narumol Prangnawarat7ddbbae2020-03-13 10:26:05 +0000[diff] [blame]63 bool IsConvertBf16ToFp32Supported(const TensorInfo& input,
64 const TensorInfo& output,
65 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
66
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]67 bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
68 const TensorInfo& output,
69 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
70
71 bool IsConvertFp32ToFp16Supported(const TensorInfo& input,
72 const TensorInfo& output,
73 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
74
75 bool IsConvolution2dSupported(const TensorInfo& input,
76 const TensorInfo& output,
77 const Convolution2dDescriptor& descriptor,
78 const TensorInfo& weights,
79 const Optional<TensorInfo>& biases,
80 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
81
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +0000[diff] [blame]82 bool IsDebugSupported(const TensorInfo& input,
83 const TensorInfo& output,
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +0000[diff] [blame]84 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
85
Aron Virginas-Tar73f66422019-09-23 19:11:59 +0100[diff] [blame]86 bool IsDepthToSpaceSupported(const TensorInfo& input,
87 const TensorInfo& output,
88 const DepthToSpaceDescriptor& descriptor,
89 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
90
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]91 bool IsDepthwiseConvolutionSupported(const TensorInfo& input,
92 const TensorInfo& output,
93 const DepthwiseConvolution2dDescriptor& descriptor,
94 const TensorInfo& weights,
95 const Optional<TensorInfo>& biases,
96 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
97
Nattapat Chaimanowong8a54ac02019-03-29 15:25:04 +0000[diff] [blame]98 bool IsDequantizeSupported(const TensorInfo& input,
99 const TensorInfo& output,
100 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
101
Derek Lamberti6a5e5e82019-12-05 14:41:20 +0000[diff] [blame]102 bool IsDetectionPostProcessSupported(const TensorInfo& boxEncodings,
103 const TensorInfo& scores,
104 const TensorInfo& anchors,
105 const TensorInfo& detectionBoxes,
106 const TensorInfo& detectionClasses,
107 const TensorInfo& detectionScores,
108 const TensorInfo& numDetections,
Narumol Prangnawaratbc67cef2019-01-31 15:31:54 +0000[diff] [blame]109 const DetectionPostProcessDescriptor& descriptor,
110 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
111
Pablo Tellof0bd6832019-04-26 17:58:13 +0100[diff] [blame]112 bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo& input,
113 const TensorInfo& output,
114 const DepthwiseConvolution2dDescriptor& descriptor,
115 const TensorInfo& weights,
116 const Optional<TensorInfo>& biases,
117 Optional<std::string&> reasonIfUnsupported =
118 EmptyOptional()) const override;
119
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]120 bool IsDivisionSupported(const TensorInfo& input0,
121 const TensorInfo& input1,
122 const TensorInfo& output,
123 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
124
josh minor4a3c6102020-01-06 16:40:46 -0600[diff] [blame]125 bool IsElementwiseUnarySupported(const TensorInfo& input,
126 const TensorInfo& output,
127 const ElementwiseUnaryDescriptor& descriptor,
128 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
129
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100[diff] [blame]130 ARMNN_DEPRECATED_MSG("Use IsComparisonSupported instead")
FrancisMurtagh30cdfca2018-12-18 12:57:35 +0000[diff] [blame]131 bool IsEqualSupported(const TensorInfo& input0,
132 const TensorInfo& input1,
133 const TensorInfo& output,
134 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
135
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]136 bool IsFakeQuantizationSupported(const TensorInfo& input,
137 const FakeQuantizationDescriptor& descriptor,
138 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
139
140 bool IsFloorSupported(const TensorInfo& input,
141 const TensorInfo& output,
142 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
143
144 bool IsFullyConnectedSupported(const TensorInfo& input,
145 const TensorInfo& output,
146 const TensorInfo& weights,
147 const TensorInfo& biases,
148 const FullyConnectedDescriptor& descriptor,
149 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
150
narpra014951d842019-01-18 16:53:53 +0000[diff] [blame]151 bool IsGatherSupported(const TensorInfo& input0,
152 const TensorInfo& input1,
153 const TensorInfo& output,
154 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
155
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100[diff] [blame]156 ARMNN_DEPRECATED_MSG("Use IsComparisonSupported instead")
FrancisMurtagh878f0232018-12-19 10:56:15 +0000[diff] [blame]157 bool IsGreaterSupported(const TensorInfo& input0,
158 const TensorInfo& input1,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]159 const TensorInfo& output,
FrancisMurtagh878f0232018-12-19 10:56:15 +0000[diff] [blame]160 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
161
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]162 bool IsInputSupported(const TensorInfo& input,
163 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
164
Kevin May09ca49c2019-10-09 12:37:34 +0100[diff] [blame]165 bool IsInstanceNormalizationSupported(const TensorInfo& input,
166 const TensorInfo& output,
167 const InstanceNormalizationDescriptor& descriptor,
168 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
169
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]170 bool IsL2NormalizationSupported(const TensorInfo& input,
171 const TensorInfo& output,
172 const L2NormalizationDescriptor& descriptor,
173 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
174
Aron Virginas-Tare662a942019-10-14 15:12:00 +0100[diff] [blame]175 bool IsLogSoftmaxSupported(const TensorInfo& input,
176 const TensorInfo& output,
177 const LogSoftmaxDescriptor& descriptor,
178 Optional<std::string&> reasonIfUnsupported) const override;
179
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]180 bool IsLstmSupported(const TensorInfo& input,
181 const TensorInfo& outputStateIn,
182 const TensorInfo& cellStateIn,
183 const TensorInfo& scratchBuffer,
184 const TensorInfo& outputStateOut,
185 const TensorInfo& cellStateOut,
186 const TensorInfo& output,
187 const LstmDescriptor& descriptor,
Jan Eilersd01a83c2019-07-03 18:20:40 +0100[diff] [blame]188 const LstmInputParamsInfo& paramsInfo,
189 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]190
saoste012df12b32018-11-28 16:57:20 +0000[diff] [blame]191 bool IsMaximumSupported(const TensorInfo& input0,
192 const TensorInfo& input1,
193 const TensorInfo& output,
194 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
195
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]196 bool IsMeanSupported(const TensorInfo& input,
197 const TensorInfo& output,
198 const MeanDescriptor& descriptor,
199 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
200
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]201 ARMNN_DEPRECATED_MSG("Use IsConcatSupported instead")
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]202 bool IsMergerSupported(const std::vector<const TensorInfo*> inputs,
Nikhil Raj8599a412018-11-19 14:51:07 +0000[diff] [blame]203 const TensorInfo& output,
Jim Flynne242f2d2019-05-22 14:24:13 +0100[diff] [blame]204 const MergerDescriptor& descriptor,
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]205 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
206
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]207 bool IsMemCopySupported(const TensorInfo& input,
208 const TensorInfo& output,
209 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
210
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000[diff] [blame]211 bool IsMinimumSupported(const TensorInfo& input0,
212 const TensorInfo& input1,
213 const TensorInfo& output,
214 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
215
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]216 bool IsMultiplicationSupported(const TensorInfo& input0,
217 const TensorInfo& input1,
218 const TensorInfo& output,
219 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
220
221 bool IsNormalizationSupported(const TensorInfo& input,
222 const TensorInfo& output,
223 const NormalizationDescriptor& descriptor,
224 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
225
226 bool IsOutputSupported(const TensorInfo& output,
227 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
228
229 bool IsPadSupported(const TensorInfo& input,
230 const TensorInfo& output,
231 const PadDescriptor& descriptor,
232 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
233
234 bool IsPermuteSupported(const TensorInfo& input,
235 const TensorInfo& output,
236 const PermuteDescriptor& descriptor,
237 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
238
239 bool IsPooling2dSupported(const TensorInfo& input,
240 const TensorInfo& output,
241 const Pooling2dDescriptor& descriptor,
242 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
243
Derek Lamberti5f400d62019-03-25 15:41:58 +0000[diff] [blame]244 bool IsQuantizeSupported(const TensorInfo& input,
245 const TensorInfo& output,
246 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
247
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]248 bool IsReshapeSupported(const TensorInfo& input,
Kevin Maya023c402019-12-12 17:28:05 +0000[diff] [blame]249 const TensorInfo& output,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]250 const ReshapeDescriptor& descriptor,
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]251 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
252
253 bool IsResizeBilinearSupported(const TensorInfo& input,
Sadik Armaganc625f002018-12-17 11:32:16 +0000[diff] [blame]254 const TensorInfo& output,
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]255 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
256
Teresa Charlin970f43b2019-07-01 13:51:07 +0100[diff] [blame]257 bool IsResizeSupported(const TensorInfo& input,
258 const TensorInfo& output,
259 const ResizeDescriptor& descriptor,
260 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
josh minor4a3c6102020-01-06 16:40:46 -0600[diff] [blame]261
262 ARMNN_DEPRECATED_MSG("Use IsElementwiseUnarySupported instead")
Mohamed Nour Abouelseouda1d3c6a2018-12-27 12:39:16 +0000[diff] [blame]263 bool IsRsqrtSupported(const TensorInfo& input,
264 const TensorInfo& output,
265 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
266
Aron Virginas-Tar92b9f872019-09-17 17:27:04 +0100[diff] [blame]267 bool IsSliceSupported(const TensorInfo& input,
268 const TensorInfo& output,
269 const SliceDescriptor& descriptor,
270 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
271
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]272 bool IsSoftmaxSupported(const TensorInfo& input,
273 const TensorInfo& output,
274 const SoftmaxDescriptor& descriptor,
275 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
276
Nattapat Chaimanowong3ea76d52018-11-09 14:10:38 +0000[diff] [blame]277 bool IsSpaceToBatchNdSupported(const TensorInfo& input,
278 const TensorInfo& output,
279 const SpaceToBatchNdDescriptor& descriptor,
280 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
281
Keith Davisa57eccb2019-06-14 17:33:22 +0100[diff] [blame]282 bool IsSpaceToDepthSupported(const TensorInfo& input,
283 const TensorInfo& output,
284 const SpaceToDepthDescriptor& descriptor,
285 Optional<std::string&> reasonIfUnsupported = EmptyOptional())
286 const override;
287
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100[diff] [blame]288 ARMNN_DEPRECATED_MSG("Use IsSplitterSupported with outputs instead")
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]289 bool IsSplitterSupported(const TensorInfo& input,
290 const ViewsDescriptor& descriptor,
291 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
292
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100[diff] [blame]293 bool IsSplitterSupported(const TensorInfo& input,
294 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
295 const ViewsDescriptor& descriptor,
296 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
297
Matthew Jackson81e601c2019-07-11 12:07:09 +0100[diff] [blame]298 bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
299 const TensorInfo& output,
300 const StackDescriptor& descriptor,
301 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
302
Nattapat Chaimanowong1216b582018-11-23 15:33:41 +0000[diff] [blame]303 bool IsStridedSliceSupported(const TensorInfo& input,
304 const TensorInfo& output,
305 const StridedSliceDescriptor& descriptor,
306 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
307
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]308 bool IsSubtractionSupported(const TensorInfo& input0,
309 const TensorInfo& input1,
310 const TensorInfo& output,
311 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Matteo Martincighab9e5252019-06-13 17:27:46 +0100[diff] [blame]312
313 bool IsPreluSupported(const TensorInfo& input,
314 const TensorInfo& alpha,
315 const TensorInfo& output,
316 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Aron Virginas-Tar98180ef2019-06-26 15:02:47 +0100[diff] [blame]317
318 bool IsTransposeConvolution2dSupported(
319 const TensorInfo& input,
320 const TensorInfo& output,
321 const TransposeConvolution2dDescriptor& descriptor,
322 const TensorInfo& weights,
323 const Optional<TensorInfo>& biases,
324 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000[diff] [blame]325
326 bool IsTransposeSupported(const TensorInfo& input,
327 const TensorInfo& output,
328 const TransposeDescriptor& descriptor,
329 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
330
arovir014424b0a2018-10-04 10:46:04 +0100[diff] [blame]331};
332
Aron Virginas-Tarb5acbb72018-10-15 11:11:51 +0100[diff] [blame]333} // namespace armnn