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