Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / e242f2dc646f41e9162aaaf74e057ce39fcb92df / . / src / backends / reference / RefLayerSupport.hpp
blob: 8850c6e105070fe7617672f278bda7a3360d6799 [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:
15 bool IsActivationSupported(const TensorInfo& input,
16 const TensorInfo& output,
17 const ActivationDescriptor& descriptor,
18 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
19
20 bool IsAdditionSupported(const TensorInfo& input0,
21 const TensorInfo& input1,
22 const TensorInfo& output,
23 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
24
25 bool IsBatchNormalizationSupported(const TensorInfo& input,
26 const TensorInfo& output,
27 const TensorInfo& mean,
28 const TensorInfo& var,
29 const TensorInfo& beta,
30 const TensorInfo& gamma,
31 const BatchNormalizationDescriptor& descriptor,
32 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
33
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000[diff] [blame]34 bool IsBatchToSpaceNdSupported(const TensorInfo& input,
35 const TensorInfo& output,
36 const BatchToSpaceNdDescriptor& descriptor,
37 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
38
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]39 bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,
40 const TensorInfo& output,
Jim Flynne242f2d2019-05-22 14:24:13 +0100[diff] [blame^]41 const ConcatDescriptor& descriptor,
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]42 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
43
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]44 bool IsConstantSupported(const TensorInfo& output,
45 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
46
47 bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
48 const TensorInfo& output,
49 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
50
51 bool IsConvertFp32ToFp16Supported(const TensorInfo& input,
52 const TensorInfo& output,
53 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
54
55 bool IsConvolution2dSupported(const TensorInfo& input,
56 const TensorInfo& output,
57 const Convolution2dDescriptor& descriptor,
58 const TensorInfo& weights,
59 const Optional<TensorInfo>& biases,
60 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
61
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +0000[diff] [blame]62 bool IsDebugSupported(const TensorInfo& input,
63 const TensorInfo& output,
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +0000[diff] [blame]64 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
65
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]66 bool IsDepthwiseConvolutionSupported(const TensorInfo& input,
67 const TensorInfo& output,
68 const DepthwiseConvolution2dDescriptor& descriptor,
69 const TensorInfo& weights,
70 const Optional<TensorInfo>& biases,
71 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
72
Nattapat Chaimanowong8a54ac02019-03-29 15:25:04 +0000[diff] [blame]73 bool IsDequantizeSupported(const TensorInfo& input,
74 const TensorInfo& output,
75 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
76
Narumol Prangnawaratbc67cef2019-01-31 15:31:54 +0000[diff] [blame]77 bool IsDetectionPostProcessSupported(const TensorInfo& input0,
78 const TensorInfo& input1,
79 const DetectionPostProcessDescriptor& descriptor,
80 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
81
Pablo Tellof0bd6832019-04-26 17:58:13 +0100[diff] [blame]82 bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo& input,
83 const TensorInfo& output,
84 const DepthwiseConvolution2dDescriptor& descriptor,
85 const TensorInfo& weights,
86 const Optional<TensorInfo>& biases,
87 Optional<std::string&> reasonIfUnsupported =
88 EmptyOptional()) const override;
89
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]90 bool IsDivisionSupported(const TensorInfo& input0,
91 const TensorInfo& input1,
92 const TensorInfo& output,
93 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
94
FrancisMurtagh30cdfca2018-12-18 12:57:35 +0000[diff] [blame]95 bool IsEqualSupported(const TensorInfo& input0,
96 const TensorInfo& input1,
97 const TensorInfo& output,
98 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
99
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]100 bool IsFakeQuantizationSupported(const TensorInfo& input,
101 const FakeQuantizationDescriptor& descriptor,
102 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
103
104 bool IsFloorSupported(const TensorInfo& input,
105 const TensorInfo& output,
106 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
107
108 bool IsFullyConnectedSupported(const TensorInfo& input,
109 const TensorInfo& output,
110 const TensorInfo& weights,
111 const TensorInfo& biases,
112 const FullyConnectedDescriptor& descriptor,
113 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
114
narpra014951d842019-01-18 16:53:53 +0000[diff] [blame]115 bool IsGatherSupported(const TensorInfo& input0,
116 const TensorInfo& input1,
117 const TensorInfo& output,
118 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
119
FrancisMurtagh878f0232018-12-19 10:56:15 +0000[diff] [blame]120 bool IsGreaterSupported(const TensorInfo& input0,
121 const TensorInfo& input1,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]122 const TensorInfo& output,
FrancisMurtagh878f0232018-12-19 10:56:15 +0000[diff] [blame]123 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
124
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]125 bool IsInputSupported(const TensorInfo& input,
126 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
127
128 bool IsL2NormalizationSupported(const TensorInfo& input,
129 const TensorInfo& output,
130 const L2NormalizationDescriptor& descriptor,
131 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
132
133 bool IsLstmSupported(const TensorInfo& input,
134 const TensorInfo& outputStateIn,
135 const TensorInfo& cellStateIn,
136 const TensorInfo& scratchBuffer,
137 const TensorInfo& outputStateOut,
138 const TensorInfo& cellStateOut,
139 const TensorInfo& output,
140 const LstmDescriptor& descriptor,
141 const TensorInfo& inputToForgetWeights,
142 const TensorInfo& inputToCellWeights,
143 const TensorInfo& inputToOutputWeights,
144 const TensorInfo& recurrentToForgetWeights,
145 const TensorInfo& recurrentToCellWeights,
146 const TensorInfo& recurrentToOutputWeights,
147 const TensorInfo& forgetGateBias,
148 const TensorInfo& cellBias,
149 const TensorInfo& outputGateBias,
150 const TensorInfo* inputToInputWeights,
151 const TensorInfo* recurrentToInputWeights,
152 const TensorInfo* cellToInputWeights,
153 const TensorInfo* inputGateBias,
154 const TensorInfo* projectionWeights,
155 const TensorInfo* projectionBias,
156 const TensorInfo* cellToForgetWeights,
157 const TensorInfo* cellToOutputWeights,
158 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
159
saoste012df12b32018-11-28 16:57:20 +0000[diff] [blame]160 bool IsMaximumSupported(const TensorInfo& input0,
161 const TensorInfo& input1,
162 const TensorInfo& output,
163 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
164
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]165 bool IsMeanSupported(const TensorInfo& input,
166 const TensorInfo& output,
167 const MeanDescriptor& descriptor,
168 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
169
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]170 ARMNN_DEPRECATED_MSG("Use IsConcatSupported instead")
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]171 bool IsMergerSupported(const std::vector<const TensorInfo*> inputs,
Nikhil Raj8599a412018-11-19 14:51:07 +0000[diff] [blame]172 const TensorInfo& output,
Jim Flynne242f2d2019-05-22 14:24:13 +0100[diff] [blame^]173 const MergerDescriptor& descriptor,
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]174 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
175
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]176 bool IsMemCopySupported(const TensorInfo& input,
177 const TensorInfo& output,
178 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
179
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000[diff] [blame]180 bool IsMinimumSupported(const TensorInfo& input0,
181 const TensorInfo& input1,
182 const TensorInfo& output,
183 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
184
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]185 bool IsMultiplicationSupported(const TensorInfo& input0,
186 const TensorInfo& input1,
187 const TensorInfo& output,
188 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
189
190 bool IsNormalizationSupported(const TensorInfo& input,
191 const TensorInfo& output,
192 const NormalizationDescriptor& descriptor,
193 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
194
195 bool IsOutputSupported(const TensorInfo& output,
196 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
197
198 bool IsPadSupported(const TensorInfo& input,
199 const TensorInfo& output,
200 const PadDescriptor& descriptor,
201 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
202
203 bool IsPermuteSupported(const TensorInfo& input,
204 const TensorInfo& output,
205 const PermuteDescriptor& descriptor,
206 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
207
208 bool IsPooling2dSupported(const TensorInfo& input,
209 const TensorInfo& output,
210 const Pooling2dDescriptor& descriptor,
211 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
212
Derek Lamberti5f400d62019-03-25 15:41:58 +0000[diff] [blame]213 bool IsQuantizeSupported(const TensorInfo& input,
214 const TensorInfo& output,
215 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
216
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]217 bool IsReshapeSupported(const TensorInfo& input,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]218 const ReshapeDescriptor& descriptor,
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]219 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
220
221 bool IsResizeBilinearSupported(const TensorInfo& input,
Sadik Armaganc625f002018-12-17 11:32:16 +0000[diff] [blame]222 const TensorInfo& output,
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]223 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
224
Mohamed Nour Abouelseouda1d3c6a2018-12-27 12:39:16 +0000[diff] [blame]225 bool IsRsqrtSupported(const TensorInfo& input,
226 const TensorInfo& output,
227 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
228
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]229 bool IsSoftmaxSupported(const TensorInfo& input,
230 const TensorInfo& output,
231 const SoftmaxDescriptor& descriptor,
232 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
233
Nattapat Chaimanowong3ea76d52018-11-09 14:10:38 +0000[diff] [blame]234 bool IsSpaceToBatchNdSupported(const TensorInfo& input,
235 const TensorInfo& output,
236 const SpaceToBatchNdDescriptor& descriptor,
237 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
238
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100[diff] [blame]239 ARMNN_DEPRECATED_MSG("Use IsSplitterSupported with outputs instead")
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]240 bool IsSplitterSupported(const TensorInfo& input,
241 const ViewsDescriptor& descriptor,
242 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
243
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100[diff] [blame]244 bool IsSplitterSupported(const TensorInfo& input,
245 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
246 const ViewsDescriptor& descriptor,
247 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
248
Nattapat Chaimanowong1216b582018-11-23 15:33:41 +0000[diff] [blame]249 bool IsStridedSliceSupported(const TensorInfo& input,
250 const TensorInfo& output,
251 const StridedSliceDescriptor& descriptor,
252 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
253
arovir011c7c81b2018-10-08 11:34:28 +0100[diff] [blame]254 bool IsSubtractionSupported(const TensorInfo& input0,
255 const TensorInfo& input1,
256 const TensorInfo& output,
257 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
arovir014424b0a2018-10-04 10:46:04 +0100[diff] [blame]258};
259
Aron Virginas-Tarb5acbb72018-10-15 11:11:51 +0100[diff] [blame]260} // namespace armnn