Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 8ac5fc39b2556fd3ac0d0c43ebf55525bde97f58 / . / src / backends / neon / NeonLayerSupport.hpp
blob: 58181d375b8ef20eac89215f938e821ec1c0d8a9 [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>
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]8
9namespace armnn
10{
11
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]12class NeonLayerSupport : public LayerSupportBase
arovir014424b0a2018-10-04 10:46:04 +0100[diff] [blame]13{
arovir017ff76c52018-10-09 09:40:58 +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
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]34 bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,
35 const TensorInfo& output,
Jim Flynne242f2d2019-05-22 14:24:13 +0100[diff] [blame]36 const ConcatDescriptor& descriptor,
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]37 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
38
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]39 bool IsConstantSupported(const TensorInfo& output,
40 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
41
42 bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
43 const TensorInfo& output,
44 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
45
46 bool IsConvertFp32ToFp16Supported(const TensorInfo& input,
47 const TensorInfo& output,
48 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
49
50 bool IsConvolution2dSupported(const TensorInfo& input,
51 const TensorInfo& output,
52 const Convolution2dDescriptor& descriptor,
53 const TensorInfo& weights,
54 const Optional<TensorInfo>& biases,
55 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
56
57 bool IsDepthwiseConvolutionSupported(const TensorInfo& input,
58 const TensorInfo& output,
59 const DepthwiseConvolution2dDescriptor& descriptor,
60 const TensorInfo& weights,
61 const Optional<TensorInfo>& biases,
62 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
63
Narumol Prangnawarat01961a72019-05-30 16:47:12 +0100[diff] [blame]64 bool IsDequantizeSupported(const TensorInfo& input,
65 const TensorInfo& output,
66 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
67
Pablo Tellof0bd6832019-04-26 17:58:13 +0100[diff] [blame]68 bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo& input,
69 const TensorInfo& output,
70 const DepthwiseConvolution2dDescriptor& descriptor,
71 const TensorInfo& weights,
72 const Optional<TensorInfo>& biases,
73 Optional<std::string&> reason = EmptyOptional()) const override;
74
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]75 bool IsFloorSupported(const TensorInfo& input,
76 const TensorInfo& output,
77 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
78
79 bool IsFullyConnectedSupported(const TensorInfo& input,
80 const TensorInfo& output,
81 const TensorInfo& weights,
82 const TensorInfo& biases,
83 const FullyConnectedDescriptor& descriptor,
84 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
85
kevmay01eed85922019-01-28 08:37:25 +0000[diff] [blame]86 bool IsGreaterSupported(const TensorInfo& input0,
87 const TensorInfo& input1,
88 const TensorInfo& output,
89 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
90
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]91 bool IsInputSupported(const TensorInfo& input,
92 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
93
94 bool IsL2NormalizationSupported(const TensorInfo& input,
95 const TensorInfo& output,
96 const L2NormalizationDescriptor& descriptor,
97 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
98
Jan Eilersad5293a2019-07-08 09:57:55 +0100[diff] [blame]99 bool IsLstmSupported(const TensorInfo& input,
100 const TensorInfo& outputStateIn,
101 const TensorInfo& cellStateIn,
102 const TensorInfo& scratchBuffer,
103 const TensorInfo& outputStateOut,
104 const TensorInfo& cellStateOut,
105 const TensorInfo& output,
106 const LstmDescriptor& descriptor,
107 const LstmInputParamsInfo& paramsInfo,
108 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
109
Nattapat Chaimanowong4e6597a2018-12-20 14:14:06 +0000[diff] [blame]110 bool IsMaximumSupported(const TensorInfo& input0,
111 const TensorInfo& input1,
112 const TensorInfo& output,
113 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
114
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]115 bool IsMeanSupported(const TensorInfo& input,
116 const TensorInfo& output,
117 const MeanDescriptor& descriptor,
118 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
119
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]120 ARMNN_DEPRECATED_MSG("Use IsConcatSupported instead")
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]121 bool IsMergerSupported(const std::vector<const TensorInfo*> inputs,
Nikhil Raj8599a412018-11-19 14:51:07 +0000[diff] [blame]122 const TensorInfo& output,
Jim Flynne242f2d2019-05-22 14:24:13 +0100[diff] [blame]123 const MergerDescriptor& descriptor,
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]124 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
125
Conor Kennedy54b21692019-01-09 07:57:38 +0000[diff] [blame]126 bool IsMinimumSupported(const TensorInfo& input0,
127 const TensorInfo& input1,
128 const TensorInfo& output,
129 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
130
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]131 bool IsMultiplicationSupported(const TensorInfo& input0,
132 const TensorInfo& input1,
133 const TensorInfo& output,
134 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
135
136 bool IsNormalizationSupported(const TensorInfo& input,
137 const TensorInfo& output,
138 const NormalizationDescriptor& descriptor,
139 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
140
141 bool IsOutputSupported(const TensorInfo& output,
142 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
143
Éanna Ó Catháin12055742019-01-25 10:01:40 +0000[diff] [blame]144 bool IsPadSupported(const TensorInfo& input,
145 const TensorInfo& output,
146 const PadDescriptor& descriptor,
147 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
148
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]149 bool IsPermuteSupported(const TensorInfo& input,
150 const TensorInfo& output,
151 const PermuteDescriptor& descriptor,
152 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
153
154 bool IsPooling2dSupported(const TensorInfo& input,
155 const TensorInfo& output,
156 const Pooling2dDescriptor& descriptor,
157 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
158
Nikhil Raj9b461482019-07-03 15:58:31 +0100[diff] [blame]159 bool IsPreluSupported(const TensorInfo& input,
160 const TensorInfo& alpha,
161 const TensorInfo& output,
162 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
163
Sadik Armaganfabc2892019-05-31 09:05:11 +0100[diff] [blame]164 bool IsQuantizeSupported(const TensorInfo& input,
165 const TensorInfo& output,
166 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
167
Francis Murtagh4fc3c482019-08-02 13:20:54 +0100[diff] [blame]168 bool IsQuantizedLstmSupported(const TensorInfo& input,
169 const TensorInfo& cellStateIn,
170 const TensorInfo& outputStateIn,
171 const TensorInfo& cellStateOut,
172 const TensorInfo& outputStateOut,
173 const QuantizedLstmInputParamsInfo& paramsInfo,
174 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
175
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]176 bool IsReshapeSupported(const TensorInfo& input,
Kevin May8ac5fc32019-12-12 17:28:05 +0000[diff] [blame^]177 const TensorInfo& output,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]178 const ReshapeDescriptor& descriptor,
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]179 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
180
Aron Virginas-Tar169d2f12019-07-01 19:01:44 +0100[diff] [blame]181 bool IsResizeSupported(const TensorInfo& input,
182 const TensorInfo& output,
183 const ResizeDescriptor& descriptor,
184 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
185
186 ARMNN_DEPRECATED_MSG("Use IsResizeSupported instead")
Sadik Armaganc625f002018-12-17 11:32:16 +0000[diff] [blame]187 bool IsResizeBilinearSupported(const TensorInfo& input,
188 const TensorInfo& output,
189 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
190
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]191 bool IsSoftmaxSupported(const TensorInfo& input,
192 const TensorInfo& output,
193 const SoftmaxDescriptor& descriptor,
194 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
195
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100[diff] [blame]196 bool IsSpaceToDepthSupported(const TensorInfo& input,
197 const TensorInfo& output,
198 const SpaceToDepthDescriptor& descriptor,
199 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
200
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100[diff] [blame]201 ARMNN_DEPRECATED_MSG("Use IsSplitterSupported with outputs instead")
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]202 bool IsSplitterSupported(const TensorInfo& input,
203 const ViewsDescriptor& descriptor,
204 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
205
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100[diff] [blame]206 bool IsSplitterSupported(const TensorInfo& input,
207 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
208 const ViewsDescriptor& descriptor,
209 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
210
Matthew Jackson87f65ea2019-08-01 10:01:34 +0100[diff] [blame]211 bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
212 const TensorInfo& output,
213 const StackDescriptor& descriptor,
214 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
215
FinnWilliamsArm1fa19192019-08-02 17:26:31 +0100[diff] [blame]216 bool IsStridedSliceSupported(const TensorInfo& input,
217 const TensorInfo& output,
218 const StridedSliceDescriptor& descriptor,
219 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
220
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]221 bool IsSubtractionSupported(const TensorInfo& input0,
222 const TensorInfo& input1,
223 const TensorInfo& output,
224 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]225
226 bool IsTransposeConvolution2dSupported(const TensorInfo& input,
227 const TensorInfo& output,
228 const TransposeConvolution2dDescriptor& descriptor,
229 const TensorInfo& weights,
230 const Optional<TensorInfo>& biases,
231 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
232
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]233}; // class NeonLayerSupport
arovir014424b0a2018-10-04 10:46:04 +0100[diff] [blame]234
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]235} // namespace armnn