Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 9145e38edf49fa4862008c163c34590141eecb14 / . / src / backends / neon / NeonLayerSupport.hpp
blob: 0295c2b3e25b8f8fdb2881cc13fe6aa9f0528969 [file] [log] [blame]
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1//
Keith Davis69e653f2020-07-02 11:49:26 +0100[diff] [blame]2// Copyright © 2017 Arm Ltd and Contributors. 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
Sadik Armagan045f6be2020-09-10 13:37:32 +0100[diff] [blame]7#include <armnn/backends/IBackendInternal.hpp>
8
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]9#include <backendsCommon/LayerSupportBase.hpp>
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]10
11namespace armnn
12{
13
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]14class NeonLayerSupport : public LayerSupportBase
arovir014424b0a2018-10-04 10:46:04 +0100[diff] [blame]15{
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]16public:
Sadik Armagan045f6be2020-09-10 13:37:32 +0100[diff] [blame]17 explicit NeonLayerSupport(const IBackendInternal::IBackendSpecificModelContextPtr& modelContextPtr);
18 NeonLayerSupport();
19
20 ~NeonLayerSupport() {}
21
Cathal Corbett34b429c2021-12-24 12:24:40 +0000[diff] [blame]22 bool IsLayerSupported(const LayerType& type,
23 const std::vector<TensorInfo>& infos,
24 const BaseDescriptor& descriptor,
25 const Optional<LstmInputParamsInfo>& lstmParamsInfo,
26 const Optional<QuantizedLstmInputParamsInfo>& quantizedLstmParamsInfo,
27 Optional<std::string&> reasonIfUnsupported) const override;
28
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]29 bool IsActivationSupported(const TensorInfo& input,
30 const TensorInfo& output,
31 const ActivationDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]32 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]33
34 bool IsAdditionSupported(const TensorInfo& input0,
35 const TensorInfo& input1,
36 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]37 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]38
James Conroyd47a0642019-09-17 14:22:06 +0100[diff] [blame]39 bool IsArgMinMaxSupported(const TensorInfo& input,
40 const TensorInfo& output,
41 const ArgMinMaxDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]42 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
James Conroyd47a0642019-09-17 14:22:06 +0100[diff] [blame]43
Teresa Charlin0f86ecf2022-10-13 15:47:08 +0100[diff] [blame]44 bool IsBatchMatMulSupported(const TensorInfo& inputX,
45 const TensorInfo& inputY,
46 const TensorInfo& output,
47 const BatchMatMulDescriptor& descriptor,
48 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
49
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]50 bool IsBatchNormalizationSupported(const TensorInfo& input,
51 const TensorInfo& output,
52 const TensorInfo& mean,
53 const TensorInfo& var,
54 const TensorInfo& beta,
55 const TensorInfo& gamma,
56 const BatchNormalizationDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]57 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]58
Mike Kelly56858022020-01-27 12:14:47 +0000[diff] [blame]59 bool IsBatchToSpaceNdSupported(const TensorInfo& input,
60 const TensorInfo& output,
61 const BatchToSpaceNdDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]62 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Mike Kelly56858022020-01-27 12:14:47 +0000[diff] [blame]63
Sadik Armagan48f011e2021-04-21 10:50:34 +0100[diff] [blame]64 bool IsCastSupported(const TensorInfo& input,
65 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]66 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Sadik Armagan48f011e2021-04-21 10:50:34 +0100[diff] [blame]67
Teresa Charline89dd692021-09-01 16:30:34 +0100[diff] [blame]68 bool IsChannelShuffleSupported(const TensorInfo& input,
69 const TensorInfo& output,
70 const ChannelShuffleDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]71 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Teresa Charline89dd692021-09-01 16:30:34 +0100[diff] [blame]72
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100[diff] [blame]73 bool IsComparisonSupported(const TensorInfo& input0,
74 const TensorInfo& input1,
75 const TensorInfo& output,
76 const ComparisonDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]77 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100[diff] [blame]78
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]79 bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,
80 const TensorInfo& output,
Cathal Corbett34b429c2021-12-24 12:24:40 +0000[diff] [blame]81 const OriginsDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]82 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Jim Flynn906f9462019-05-10 13:55:21 +0100[diff] [blame]83
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]84 bool IsConstantSupported(const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]85 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]86
87 bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
88 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]89 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]90
91 bool IsConvertFp32ToFp16Supported(const TensorInfo& input,
92 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]93 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]94
95 bool IsConvolution2dSupported(const TensorInfo& input,
96 const TensorInfo& output,
97 const Convolution2dDescriptor& descriptor,
98 const TensorInfo& weights,
99 const Optional<TensorInfo>& biases,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]100 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]101
Teresa Charlinec5f7d12021-10-22 17:15:00 +0100[diff] [blame]102 bool IsConvolution3dSupported(const TensorInfo& input,
103 const TensorInfo& output,
104 const Convolution3dDescriptor& descriptor,
105 const TensorInfo& weights,
106 const Optional<TensorInfo>& biases,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]107 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Teresa Charlinec5f7d12021-10-22 17:15:00 +0100[diff] [blame]108
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100[diff] [blame]109 bool IsDepthToSpaceSupported(const TensorInfo& input,
110 const TensorInfo& output,
111 const DepthToSpaceDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]112 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100[diff] [blame]113
114
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]115 bool IsDepthwiseConvolutionSupported(const TensorInfo& input,
116 const TensorInfo& output,
117 const DepthwiseConvolution2dDescriptor& descriptor,
118 const TensorInfo& weights,
119 const Optional<TensorInfo>& biases,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]120 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]121
Narumol Prangnawarat01961a72019-05-30 16:47:12 +0100[diff] [blame]122 bool IsDequantizeSupported(const TensorInfo& input,
123 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]124 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Narumol Prangnawarat01961a72019-05-30 16:47:12 +0100[diff] [blame]125
Pablo Tellof0bd6832019-04-26 17:58:13 +0100[diff] [blame]126 bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo& input,
127 const TensorInfo& output,
128 const DepthwiseConvolution2dDescriptor& descriptor,
129 const TensorInfo& weights,
130 const Optional<TensorInfo>& biases,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]131 Optional<std::string&> reason = EmptyOptional()) const;
Pablo Tellof0bd6832019-04-26 17:58:13 +0100[diff] [blame]132
josh minor4a3c6102020-01-06 16:40:46 -0600[diff] [blame]133 bool IsElementwiseUnarySupported(const TensorInfo& input,
134 const TensorInfo& output,
135 const ElementwiseUnaryDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]136 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
josh minor4a3c6102020-01-06 16:40:46 -0600[diff] [blame]137
Sadik Armagana792a052020-06-23 16:22:23 +0100[diff] [blame]138 bool IsFillSupported(const TensorInfo& input,
139 const TensorInfo& output,
140 const FillDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]141 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Sadik Armagana792a052020-06-23 16:22:23 +0100[diff] [blame]142
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]143 bool IsFloorSupported(const TensorInfo& input,
144 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]145 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]146
147 bool IsFullyConnectedSupported(const TensorInfo& input,
148 const TensorInfo& output,
149 const TensorInfo& weights,
150 const TensorInfo& biases,
151 const FullyConnectedDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]152 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]153
Teresa Charlin9145e382023-08-17 18:44:58 +0100[diff] [blame^]154 bool IsFusedSupported(const std::vector<std::reference_wrapper<TensorInfo>>& inputs,
155 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
156 const FusedDescriptor& descriptor,
157 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
158
Teresa Charlinbd22c7d2022-04-26 18:14:12 +0100[diff] [blame]159 bool IsGatherNdSupported(const TensorInfo& input0,
160 const TensorInfo& input1,
161 const TensorInfo& output,
162 Optional<std::string&> reasonIfUnsupported) const;
163
Teresa Charlinf540eb82020-04-10 19:24:55 +0100[diff] [blame]164 bool IsGatherSupported(const TensorInfo& input0,
165 const TensorInfo& input1,
166 const TensorInfo& output,
Teresa Charlin52664732020-06-29 16:27:03 +0100[diff] [blame]167 const GatherDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]168 Optional<std::string&> reasonIfUnsupported) const;
Teresa Charlinf540eb82020-04-10 19:24:55 +0100[diff] [blame]169
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]170 bool IsInputSupported(const TensorInfo& input,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]171 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]172
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100[diff] [blame]173 bool IsInstanceNormalizationSupported(const TensorInfo& input,
174 const TensorInfo& output,
175 const InstanceNormalizationDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]176 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100[diff] [blame]177
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]178 bool IsL2NormalizationSupported(const TensorInfo& input,
179 const TensorInfo& output,
180 const L2NormalizationDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]181 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]182
James Conroy177df1e2020-11-13 10:18:51 +0000[diff] [blame]183 bool IsLogicalBinarySupported(const TensorInfo& input0,
184 const TensorInfo& input1,
185 const TensorInfo& output,
186 const LogicalBinaryDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]187 Optional<std::string&> reasonIfUnsupported) const;
James Conroy177df1e2020-11-13 10:18:51 +0000[diff] [blame]188
Keith Davis69e653f2020-07-02 11:49:26 +0100[diff] [blame]189 bool IsLogSoftmaxSupported(const TensorInfo& input,
190 const TensorInfo& output,
191 const LogSoftmaxDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]192 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Keith Davis69e653f2020-07-02 11:49:26 +0100[diff] [blame]193
Jan Eilersad5293a2019-07-08 09:57:55 +0100[diff] [blame]194 bool IsLstmSupported(const TensorInfo& input,
195 const TensorInfo& outputStateIn,
196 const TensorInfo& cellStateIn,
197 const TensorInfo& scratchBuffer,
198 const TensorInfo& outputStateOut,
199 const TensorInfo& cellStateOut,
200 const TensorInfo& output,
201 const LstmDescriptor& descriptor,
202 const LstmInputParamsInfo& paramsInfo,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]203 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Jan Eilersad5293a2019-07-08 09:57:55 +0100[diff] [blame]204
Nattapat Chaimanowong4e6597a2018-12-20 14:14:06 +0000[diff] [blame]205 bool IsMaximumSupported(const TensorInfo& input0,
206 const TensorInfo& input1,
207 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]208 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Nattapat Chaimanowong4e6597a2018-12-20 14:14:06 +0000[diff] [blame]209
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]210 bool IsMeanSupported(const TensorInfo& input,
211 const TensorInfo& output,
212 const MeanDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]213 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]214
Conor Kennedy54b21692019-01-09 07:57:38 +0000[diff] [blame]215 bool IsMinimumSupported(const TensorInfo& input0,
216 const TensorInfo& input1,
217 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]218 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Conor Kennedy54b21692019-01-09 07:57:38 +0000[diff] [blame]219
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]220 bool IsMultiplicationSupported(const TensorInfo& input0,
221 const TensorInfo& input1,
222 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]223 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]224
Pablo Telloe61f0712020-01-23 10:37:17 +0000[diff] [blame]225 bool IsDivisionSupported(const TensorInfo& input0,
226 const TensorInfo& input1,
227 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]228 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Pablo Telloe61f0712020-01-23 10:37:17 +0000[diff] [blame]229
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]230 bool IsNormalizationSupported(const TensorInfo& input,
231 const TensorInfo& output,
232 const NormalizationDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]233 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]234
235 bool IsOutputSupported(const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]236 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]237
Éanna Ó Catháin12055742019-01-25 10:01:40 +0000[diff] [blame]238 bool IsPadSupported(const TensorInfo& input,
239 const TensorInfo& output,
240 const PadDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]241 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Éanna Ó Catháin12055742019-01-25 10:01:40 +0000[diff] [blame]242
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]243 bool IsPermuteSupported(const TensorInfo& input,
244 const TensorInfo& output,
245 const PermuteDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]246 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]247
248 bool IsPooling2dSupported(const TensorInfo& input,
249 const TensorInfo& output,
250 const Pooling2dDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]251 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Ryan OShea19e79422022-05-04 00:38:03 +0100[diff] [blame]252 bool IsPooling3dSupported(const TensorInfo& input,
253 const TensorInfo& output,
254 const Pooling3dDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]255 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Nikhil Raj9b461482019-07-03 15:58:31 +0100[diff] [blame]256 bool IsPreluSupported(const TensorInfo& input,
257 const TensorInfo& alpha,
258 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]259 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Nikhil Raj9b461482019-07-03 15:58:31 +0100[diff] [blame]260
James Conroycc340932020-05-12 18:08:52 +0100[diff] [blame]261 bool IsQLstmSupported(const TensorInfo& input,
262 const TensorInfo& previousOutputIn,
263 const TensorInfo& previousCellStateIn,
264 const TensorInfo& outputStateOut,
265 const TensorInfo& cellStateOut,
266 const TensorInfo& output,
267 const QLstmDescriptor& descriptor,
268 const LstmInputParamsInfo& paramsInfo,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]269 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
James Conroycc340932020-05-12 18:08:52 +0100[diff] [blame]270
Sadik Armaganfabc2892019-05-31 09:05:11 +0100[diff] [blame]271 bool IsQuantizeSupported(const TensorInfo& input,
272 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]273 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Sadik Armaganfabc2892019-05-31 09:05:11 +0100[diff] [blame]274
Francis Murtagh4fc3c482019-08-02 13:20:54 +0100[diff] [blame]275 bool IsQuantizedLstmSupported(const TensorInfo& input,
276 const TensorInfo& cellStateIn,
277 const TensorInfo& outputStateIn,
278 const TensorInfo& cellStateOut,
279 const TensorInfo& outputStateOut,
280 const QuantizedLstmInputParamsInfo& paramsInfo,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]281 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Francis Murtagh4fc3c482019-08-02 13:20:54 +0100[diff] [blame]282
Sadik Armagana2747482021-02-09 10:28:54 +0000[diff] [blame]283 bool IsReduceSupported(const TensorInfo& input,
284 const TensorInfo& output,
285 const ReduceDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]286 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Sadik Armagana2747482021-02-09 10:28:54 +0000[diff] [blame]287
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]288 bool IsReshapeSupported(const TensorInfo& input,
Kevin Maya023c402019-12-12 17:28:05 +0000[diff] [blame]289 const TensorInfo& output,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]290 const ReshapeDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]291 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]292
Aron Virginas-Tar169d2f12019-07-01 19:01:44 +0100[diff] [blame]293 bool IsResizeSupported(const TensorInfo& input,
294 const TensorInfo& output,
295 const ResizeDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]296 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Aron Virginas-Tar169d2f12019-07-01 19:01:44 +0100[diff] [blame]297
josh minor036f02d2019-11-15 14:53:22 -0600[diff] [blame]298 bool IsSliceSupported(const TensorInfo& input,
299 const TensorInfo& output,
300 const SliceDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]301 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
josh minor036f02d2019-11-15 14:53:22 -0600[diff] [blame]302
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]303 bool IsSoftmaxSupported(const TensorInfo& input,
304 const TensorInfo& output,
305 const SoftmaxDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]306 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]307
Mike Kelly0be3a882020-01-24 11:27:50 +0000[diff] [blame]308 bool IsSpaceToBatchNdSupported(const TensorInfo& input,
309 const TensorInfo& output,
310 const SpaceToBatchNdDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]311 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Mike Kelly0be3a882020-01-24 11:27:50 +0000[diff] [blame]312
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100[diff] [blame]313 bool IsSpaceToDepthSupported(const TensorInfo& input,
314 const TensorInfo& output,
315 const SpaceToDepthDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]316 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100[diff] [blame]317
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100[diff] [blame]318 bool IsSplitterSupported(const TensorInfo& input,
319 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
320 const ViewsDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]321 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +0100[diff] [blame]322
Matthew Jackson87f65ea2019-08-01 10:01:34 +0100[diff] [blame]323 bool IsStackSupported(const std::vector<const TensorInfo*>& inputs,
324 const TensorInfo& output,
325 const StackDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]326 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Matthew Jackson87f65ea2019-08-01 10:01:34 +0100[diff] [blame]327
FinnWilliamsArm1fa19192019-08-02 17:26:31 +0100[diff] [blame]328 bool IsStridedSliceSupported(const TensorInfo& input,
329 const TensorInfo& output,
330 const StridedSliceDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]331 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
FinnWilliamsArm1fa19192019-08-02 17:26:31 +0100[diff] [blame]332
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]333 bool IsSubtractionSupported(const TensorInfo& input0,
334 const TensorInfo& input1,
335 const TensorInfo& output,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]336 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]337
David Monahan36e6eae2023-07-26 18:37:45 +0100[diff] [blame]338 bool IsTileSupported(const TensorInfo& input0,
339 const TensorInfo& output,
340 const TileDescriptor& descriptor,
341 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
342
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]343 bool IsTransposeConvolution2dSupported(const TensorInfo& input,
344 const TensorInfo& output,
345 const TransposeConvolution2dDescriptor& descriptor,
346 const TensorInfo& weights,
347 const Optional<TensorInfo>& biases,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]348 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]349
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000[diff] [blame]350 bool IsTransposeSupported(const TensorInfo& input,
351 const TensorInfo& output,
352 const TransposeDescriptor& descriptor,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]353 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000[diff] [blame]354
Cathal Corbettfd5bec42022-03-03 15:13:23 +0000[diff] [blame]355 bool IsUnidirectionalSequenceLstmSupported(const TensorInfo& input,
356 const TensorInfo& outputStateIn,
357 const TensorInfo& cellStateIn,
Mike Kelly12994962022-04-21 11:57:09 +0100[diff] [blame]358 const TensorInfo& outputStateOut,
359 const TensorInfo& cellStateOut,
Cathal Corbettfd5bec42022-03-03 15:13:23 +0000[diff] [blame]360 const TensorInfo& output,
Cathal Corbettfd5bec42022-03-03 15:13:23 +0000[diff] [blame]361 const UnidirectionalSequenceLstmDescriptor& descriptor,
362 const LstmInputParamsInfo& paramsInfo,
Francis Murtagh66277032023-07-28 13:01:32 +0100[diff] [blame]363 Optional<std::string&> reasonIfUnsupported) const;
Cathal Corbettfd5bec42022-03-03 15:13:23 +0000[diff] [blame]364
Sadik Armagan045f6be2020-09-10 13:37:32 +0100[diff] [blame]365private:
366 const IBackendInternal::IBackendSpecificModelContextPtr m_ModelContextPtr;
367
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]368}; // class NeonLayerSupport
arovir014424b0a2018-10-04 10:46:04 +0100[diff] [blame]369
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]370} // namespace armnn