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