blob: c5fc9d0fe25880d27cdad9e5c6b47cfccfc8d1db [file] [log] [blame]
Laurent Carlier749294b2020-06-01 09:03:17 +01001//
Teresa Charlin52664732020-06-29 16:27:03 +01002// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
David Beckecb56cd2018-09-05 12:52:57 +01003// SPDX-License-Identifier: MIT
telsoa014fcda012018-03-09 14:13:49 +00004//
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +00005
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +00006#include <Layer.hpp>
7#include <LayersFwd.hpp>
David Beckdcb751f2018-10-03 11:42:42 +01008
David Beckb4540be2018-09-24 13:18:27 +01009#include <armnn/Types.hpp>
10#include <armnn/LayerSupport.hpp>
Francis Murtaghcae45682021-04-26 10:07:49 +010011#include <armnn/backends/ILayerSupport.hpp>
Sadik Armaganf0a6dec2021-03-25 07:46:55 +000012#include <armnn/BackendHelper.hpp>
Matteo Martincighc601aa62019-10-29 15:03:22 +000013#include <armnn/BackendRegistry.hpp>
Jan Eilersbb446e52020-04-02 13:56:54 +010014#include <armnn/utility/PolymorphicDowncast.hpp>
Finn Williams3e54d032020-10-22 16:53:35 +010015#include <armnn/utility/TransformIterator.hpp>
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +000016
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +000017#include <backendsCommon/WorkloadFactory.hpp>
James Conroy1f58f032021-04-27 17:13:27 +010018#include <backendsCommon/TensorHandle.hpp>
Matteo Martincighe5b8eb92019-11-28 15:45:42 +000019
Francis Murtagh46c09d02019-05-28 08:15:28 +010020#include <backendsCommon/test/WorkloadTestUtils.hpp>
telsoa014fcda012018-03-09 14:13:49 +000021
David Beck111b5d92018-11-12 14:59:37 +000022#include <sstream>
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +000023
telsoa014fcda012018-03-09 14:13:49 +000024namespace armnn
25{
26
telsoa01c577f2c2018-08-31 09:22:23 +010027namespace
28{
Finn Williams3e54d032020-10-22 16:53:35 +010029using LayerList = std::list<Layer*>;
30using Iterator = LayerList::const_iterator; // Const so pointers in the list can't be modified externally.
telsoa01c577f2c2018-08-31 09:22:23 +010031
David Beck29c75de2018-10-23 13:35:58 +010032const TensorInfo OverrideDataType(const TensorInfo& info, Optional<DataType> type)
33{
34 if (!type)
35 {
36 return info;
telsoa01c577f2c2018-08-31 09:22:23 +010037 }
38
David Beck29c75de2018-10-23 13:35:58 +010039 return TensorInfo(info.GetShape(), type.value(), info.GetQuantizationScale(), info.GetQuantizationOffset());
telsoa01c577f2c2018-08-31 09:22:23 +010040}
41
David Beck29c75de2018-10-23 13:35:58 +010042} // anonymous namespace
43
Sadik Armagan045f6be2020-09-10 13:37:32 +010044bool IWorkloadFactory::IsLayerConfigurationSupported(const BackendId& backendId,
45 const IConnectableLayer& connectableLayer,
46 Optional<DataType> dataType,
47 std::string& outReasonIfUnsupported,
48 const ModelOptions& modelOptions)
telsoa014fcda012018-03-09 14:13:49 +000049{
David Beck33f0ae02018-10-18 15:13:56 +010050 Optional<std::string&> reason = outReasonIfUnsupported;
telsoa014fcda012018-03-09 14:13:49 +000051 bool result;
Jan Eilersbb446e52020-04-02 13:56:54 +010052 const Layer& layer = *(PolymorphicDowncast<const Layer*>(&connectableLayer));
David Beckdcb751f2018-10-03 11:42:42 +010053
David Beck111b5d92018-11-12 14:59:37 +000054 auto const& backendRegistry = BackendRegistryInstance();
55 if (!backendRegistry.IsBackendRegistered(backendId))
56 {
57 std::stringstream ss;
58 ss << connectableLayer.GetName() << " is not supported on " << backendId
59 << " because this backend is not registered.";
60
61 outReasonIfUnsupported = ss.str();
62 return false;
63 }
64
65 auto backendFactory = backendRegistry.GetFactory(backendId);
66 auto backendObject = backendFactory();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +000067 auto layerSupportObject = LayerSupportHandle(backendObject->GetLayerSupport(modelOptions), backendId);
David Beck33f0ae02018-10-18 15:13:56 +010068
telsoa014fcda012018-03-09 14:13:49 +000069 switch(layer.GetType())
70 {
71 case LayerType::Activation:
72 {
Jan Eilersbb446e52020-04-02 13:56:54 +010073 auto cLayer = PolymorphicDowncast<const ActivationLayer*>(&layer);
telsoa014fcda012018-03-09 14:13:49 +000074 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
telsoa01c577f2c2018-08-31 09:22:23 +010075 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +000076 result = layerSupportObject.IsActivationSupported(
telsoa01c577f2c2018-08-31 09:22:23 +010077 OverrideDataType(input, dataType),
78 OverrideDataType(output, dataType),
79 cLayer->GetParameters(),
David Beck33f0ae02018-10-18 15:13:56 +010080 reason);
telsoa014fcda012018-03-09 14:13:49 +000081 break;
82 }
83 case LayerType::Addition:
84 {
85 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
86 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
87 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +000088 result = layerSupportObject.IsAdditionSupported(
telsoa01c577f2c2018-08-31 09:22:23 +010089 OverrideDataType(input0, dataType),
90 OverrideDataType(input1, dataType),
91 OverrideDataType(output, dataType),
David Beck33f0ae02018-10-18 15:13:56 +010092 reason);
telsoa014fcda012018-03-09 14:13:49 +000093 break;
94 }
Nikhil Rajee391d52019-09-05 17:50:44 +010095 case LayerType::ArgMinMax:
96 {
Jan Eilersbb446e52020-04-02 13:56:54 +010097 auto cLayer = PolymorphicDowncast<const ArgMinMaxLayer*>(&layer);
Nikhil Rajee391d52019-09-05 17:50:44 +010098 const ArgMinMaxDescriptor& descriptor = cLayer->GetParameters();
99
100 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
101 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000102 result = layerSupportObject.IsArgMinMaxSupported(
Nikhil Rajee391d52019-09-05 17:50:44 +0100103 OverrideDataType(input, dataType),
Narumol Prangnawaratd1f57732019-10-31 14:24:02 +0000104 OverrideDataType(output, DataType::Signed32),
Nikhil Rajee391d52019-09-05 17:50:44 +0100105 descriptor,
106 reason);
107 break;
108 }
telsoa014fcda012018-03-09 14:13:49 +0000109 case LayerType::BatchNormalization:
110 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100111 auto cLayer = PolymorphicDowncast<const BatchNormalizationLayer*>(&layer);
telsoa014fcda012018-03-09 14:13:49 +0000112 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
telsoa01c577f2c2018-08-31 09:22:23 +0100113 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
114 const TensorInfo& mean = cLayer->m_Mean->GetTensorInfo();
115 const TensorInfo& var = cLayer->m_Variance->GetTensorInfo();
116 const TensorInfo& beta = cLayer->m_Beta->GetTensorInfo();
117 const TensorInfo& gamma = cLayer->m_Gamma->GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000118 result = layerSupportObject.IsBatchNormalizationSupported(
telsoa01c577f2c2018-08-31 09:22:23 +0100119 OverrideDataType(input, dataType),
120 OverrideDataType(output, dataType),
121 OverrideDataType(mean, dataType),
122 OverrideDataType(var, dataType),
123 OverrideDataType(beta, dataType),
124 OverrideDataType(gamma, dataType),
125 cLayer->GetParameters(),
David Beck33f0ae02018-10-18 15:13:56 +0100126 reason);
telsoa014fcda012018-03-09 14:13:49 +0000127 break;
128 }
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000129 case LayerType::BatchToSpaceNd:
130 {
131 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
132 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Jan Eilersbb446e52020-04-02 13:56:54 +0100133 auto cLayer = PolymorphicDowncast<const BatchToSpaceNdLayer*>(&layer);
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000134
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000135 result = layerSupportObject.IsBatchToSpaceNdSupported(OverrideDataType(input, dataType),
136 OverrideDataType(output, dataType),
137 cLayer->GetParameters(),
138 reason);
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000139 break;
140 }
mathad01b392e982021-04-07 12:07:30 +0100141 case LayerType::Cast:
142 {
143 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
144 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
145
146 result = layerSupportObject.IsCastSupported(OverrideDataType(input, dataType),
147 OverrideDataType(output, dataType),
148 reason);
149 break;
150 }
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100151 case LayerType::Comparison:
152 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100153 auto cLayer = PolymorphicDowncast<const ComparisonLayer*>(&layer);
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100154
155 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
156 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
157 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
158
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000159 result = layerSupportObject.IsComparisonSupported(OverrideDataType(input0, dataType),
160 OverrideDataType(input1, dataType),
161 OverrideDataType(output, DataType::Boolean),
162 cLayer->GetParameters(),
163 reason);
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +0100164 break;
165 }
telsoa014fcda012018-03-09 14:13:49 +0000166 case LayerType::Constant:
167 {
168 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000169 result = layerSupportObject.IsConstantSupported(OverrideDataType(output, dataType), reason);
telsoa01c577f2c2018-08-31 09:22:23 +0100170 break;
171 }
Narumol Prangnawarat7ddbbae2020-03-13 10:26:05 +0000172 case LayerType::ConvertBf16ToFp32:
173 {
174 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
175 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000176 result = layerSupportObject.IsConvertBf16ToFp32Supported(input, output, reason);
Narumol Prangnawarat7ddbbae2020-03-13 10:26:05 +0000177 break;
178 }
telsoa01c577f2c2018-08-31 09:22:23 +0100179 case LayerType::ConvertFp16ToFp32:
180 {
181 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
182 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000183 result = layerSupportObject.IsConvertFp16ToFp32Supported(input, output, reason);
telsoa01c577f2c2018-08-31 09:22:23 +0100184 break;
185 }
Narumol Prangnawaratea54a012020-03-16 16:36:10 +0000186 case LayerType::ConvertFp32ToBf16:
187 {
188 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
189 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000190 result = layerSupportObject.IsConvertFp32ToBf16Supported(input, output, reason);
Narumol Prangnawaratea54a012020-03-16 16:36:10 +0000191 break;
192 }
telsoa01c577f2c2018-08-31 09:22:23 +0100193 case LayerType::ConvertFp32ToFp16:
194 {
195 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
196 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000197 result = layerSupportObject.IsConvertFp32ToFp16Supported(input, output, reason);
telsoa014fcda012018-03-09 14:13:49 +0000198 break;
199 }
200 case LayerType::Convolution2d:
201 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100202 auto cLayer = PolymorphicDowncast<const Convolution2dLayer*>(&layer);
arovir01a6824102018-08-28 17:40:45 +0100203
204 const TensorInfo input = OverrideDataType(layer.GetInputSlot(0).GetConnection()->GetTensorInfo(),
205 dataType);
telsoa01c577f2c2018-08-31 09:22:23 +0100206 const TensorInfo output = OverrideDataType(layer.GetOutputSlot(0).GetTensorInfo(), dataType);
Narumol Prangnawaratac2770a2020-04-01 16:51:23 +0100207 ARMNN_ASSERT(cLayer->m_Weight.get() != nullptr);
surmeh013537c2c2018-05-18 16:31:43 +0100208
arovir01a6824102018-08-28 17:40:45 +0100209 const Convolution2dDescriptor& descriptor = cLayer->GetParameters();
surmeh013537c2c2018-05-18 16:31:43 +0100210
arovir01a6824102018-08-28 17:40:45 +0100211 // Construct optional biases object based on the value of m_BiasEnabled
David Beck5eec11d2018-10-04 15:43:17 +0100212 Optional<TensorInfo> biases;
surmeh013537c2c2018-05-18 16:31:43 +0100213 if (descriptor.m_BiasEnabled)
214 {
David Beck5eec11d2018-10-04 15:43:17 +0100215 biases =
216 OverrideDataType(cLayer->m_Bias->GetTensorInfo(), GetBiasTypeFromWeightsType(dataType));
surmeh013537c2c2018-05-18 16:31:43 +0100217 }
218
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000219 result = layerSupportObject.IsConvolution2dSupported(
surmeh013537c2c2018-05-18 16:31:43 +0100220 input,
221 output,
222 descriptor,
telsoa01c577f2c2018-08-31 09:22:23 +0100223 OverrideDataType(cLayer->m_Weight->GetTensorInfo(), dataType),
arovir01a6824102018-08-28 17:40:45 +0100224 biases,
David Beck33f0ae02018-10-18 15:13:56 +0100225 reason);
telsoa014fcda012018-03-09 14:13:49 +0000226 break;
227 }
Nattapat Chaimanowonga9a1cf12018-12-03 16:06:49 +0000228 case LayerType::Debug:
229 {
230 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
231 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
232
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000233 result = layerSupportObject.IsDebugSupported(OverrideDataType(input, dataType),
Nattapat Chaimanowonga9a1cf12018-12-03 16:06:49 +0000234 OverrideDataType(output, dataType),
235 reason);
236 break;
237 }
Aron Virginas-Tardd6247f2019-09-19 14:31:17 +0100238 case LayerType::DepthToSpace:
239 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100240 auto cLayer = PolymorphicDowncast<const DepthToSpaceLayer*>(&layer);
Aron Virginas-Tardd6247f2019-09-19 14:31:17 +0100241
242 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
243 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
244
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000245 result = layerSupportObject.IsDepthToSpaceSupported(OverrideDataType(input, dataType),
Aron Virginas-Tardd6247f2019-09-19 14:31:17 +0100246 OverrideDataType(output, dataType),
247 cLayer->GetParameters(),
248 reason);
249 break;
250 }
telsoa014fcda012018-03-09 14:13:49 +0000251 case LayerType::DepthwiseConvolution2d:
252 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100253 auto cLayer = PolymorphicDowncast<const DepthwiseConvolution2dLayer*>(&layer);
telsoa01c577f2c2018-08-31 09:22:23 +0100254 const TensorInfo& input = OverrideDataType(layer.GetInputSlot(0).GetConnection()->GetTensorInfo(),
255 dataType);
256 const TensorInfo& output = OverrideDataType(layer.GetOutputSlot(0).GetTensorInfo(), dataType);
Narumol Prangnawaratac2770a2020-04-01 16:51:23 +0100257 ARMNN_ASSERT(cLayer->m_Weight.get() != nullptr);
telsoa01c577f2c2018-08-31 09:22:23 +0100258
telsoa01c577f2c2018-08-31 09:22:23 +0100259 const DepthwiseConvolution2dDescriptor& descriptor = cLayer->GetParameters();
arovir01a6824102018-08-28 17:40:45 +0100260
261 // Construct optional biases object based on the value of m_BiasEnabled
David Beck5eec11d2018-10-04 15:43:17 +0100262 Optional<TensorInfo> biases;
telsoa01c577f2c2018-08-31 09:22:23 +0100263 if (descriptor.m_BiasEnabled)
264 {
David Beck5eec11d2018-10-04 15:43:17 +0100265 biases =
266 OverrideDataType(cLayer->m_Bias->GetTensorInfo(), GetBiasTypeFromWeightsType(dataType));
telsoa01c577f2c2018-08-31 09:22:23 +0100267 }
telsoa01c577f2c2018-08-31 09:22:23 +0100268
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000269 result = layerSupportObject.IsDepthwiseConvolutionSupported(
telsoa01c577f2c2018-08-31 09:22:23 +0100270 input,
271 output,
272 descriptor,
273 OverrideDataType(cLayer->m_Weight->GetTensorInfo(), dataType),
arovir01a6824102018-08-28 17:40:45 +0100274 biases,
David Beck33f0ae02018-10-18 15:13:56 +0100275 reason);
telsoa014fcda012018-03-09 14:13:49 +0000276 break;
277 }
Nattapat Chaimanowonge4294fd2019-03-28 09:56:53 +0000278 case LayerType::Dequantize:
279 {
280 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
281 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
282
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000283 result = layerSupportObject.IsDequantizeSupported(input,
284 OverrideDataType(output, dataType),
285 reason);
Nattapat Chaimanowonge4294fd2019-03-28 09:56:53 +0000286 break;
287 }
Narumol Prangnawarat94dd5d82019-01-23 18:06:26 +0000288 case LayerType::DetectionPostProcess:
289 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100290 auto cLayer = PolymorphicDowncast<const DetectionPostProcessLayer*>(&layer);
Derek Lamberti6a5e5e82019-12-05 14:41:20 +0000291 const TensorInfo& boxEncodings = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
292 const TensorInfo& scores = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
293 const TensorInfo& anchors = cLayer->m_Anchors->GetTensorInfo();
294
295 const TensorInfo& detectionBoxes = layer.GetOutputSlot(0).GetTensorInfo();
296 const TensorInfo& detectionClasses = layer.GetOutputSlot(1).GetTensorInfo();
297 const TensorInfo& detectionScores = layer.GetOutputSlot(2).GetTensorInfo();
298 const TensorInfo& numDetections = layer.GetOutputSlot(3).GetTensorInfo();
299
Narumol Prangnawarat94dd5d82019-01-23 18:06:26 +0000300 const DetectionPostProcessDescriptor& descriptor = cLayer->GetParameters();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000301 result = layerSupportObject.IsDetectionPostProcessSupported(boxEncodings,
302 scores,
303 anchors,
304 detectionBoxes,
305 detectionClasses,
306 detectionScores,
307 numDetections,
308 descriptor,
309 reason);
Narumol Prangnawarat94dd5d82019-01-23 18:06:26 +0000310 break;
311 }
josh minor4a3c6102020-01-06 16:40:46 -0600312 case LayerType::ElementwiseUnary:
313 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100314 auto cLayer = PolymorphicDowncast<const ElementwiseUnaryLayer*>(&layer);
josh minor4a3c6102020-01-06 16:40:46 -0600315
316 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
317 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
318
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000319 result = layerSupportObject.IsElementwiseUnarySupported(OverrideDataType(input, dataType),
320 OverrideDataType(output, dataType),
321 cLayer->GetParameters(),
322 reason);
josh minor4a3c6102020-01-06 16:40:46 -0600323 break;
324 }
Ryan OSheaec6c6802020-06-05 17:17:06 +0100325 case LayerType::Fill:
326 {
327 auto cLayer = PolymorphicDowncast<const FillLayer*>(&layer);
328 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
329 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
330 const FillDescriptor& descriptor = cLayer->GetParameters();
331
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000332 result = layerSupportObject.IsFillSupported(
Ryan OSheaec6c6802020-06-05 17:17:06 +0100333 OverrideDataType(input, dataType),
334 OverrideDataType(output, dataType),
335 descriptor,
336 reason);
337 break;
338 }
telsoa014fcda012018-03-09 14:13:49 +0000339 case LayerType::FakeQuantization:
340 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100341 auto cLayer = PolymorphicDowncast<const FakeQuantizationLayer*>(&layer);
telsoa014fcda012018-03-09 14:13:49 +0000342 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000343 result = layerSupportObject.IsFakeQuantizationSupported(OverrideDataType(input, dataType),
344 cLayer->GetParameters(),
345 reason);
telsoa014fcda012018-03-09 14:13:49 +0000346 break;
347 }
348 case LayerType::Floor:
349 {
350 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
351 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000352 result = layerSupportObject.IsFloorSupported(OverrideDataType(input, dataType),
353 OverrideDataType(output, dataType),
354 reason);
telsoa014fcda012018-03-09 14:13:49 +0000355 break;
356 }
357 case LayerType::FullyConnected:
358 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100359 auto cLayer = PolymorphicDowncast<const FullyConnectedLayer*>(&layer);
telsoa014fcda012018-03-09 14:13:49 +0000360 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
telsoa01c577f2c2018-08-31 09:22:23 +0100361 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000362
363 const FullyConnectedDescriptor& descriptor = cLayer->GetParameters();
364 TensorInfo weightsInfo;
365 const TensorInfo* weightsInfoPtr = nullptr;
366
367 if (descriptor.m_ConstantWeights)
368 {
369 ARMNN_ASSERT(cLayer->m_Weight.get() != nullptr);
370 weightsInfo = OverrideDataType(cLayer->m_Weight->GetTensorInfo(), dataType);
371 }
372 else
373 {
374 weightsInfo = OverrideDataType(layer.GetInputSlot(1).GetConnection()->GetTensorInfo(), dataType);
375
376 }
377 weightsInfoPtr = &weightsInfo;
telsoa01c577f2c2018-08-31 09:22:23 +0100378
379 TensorInfo biasInfo;
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000380 const TensorInfo* biasInfoPtr = nullptr;
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000381 static const TensorInfo dummyBFloat16Bias(TensorShape({1,1,1,1}), DataType::BFloat16);
telsoa01c577f2c2018-08-31 09:22:23 +0100382 static const TensorInfo dummyFloat16Bias(TensorShape({1,1,1,1}), DataType::Float16);
383 static const TensorInfo dummyFloat32Bias(TensorShape({1,1,1,1}), DataType::Float32);
384 static const TensorInfo dummyQA8Bias(TensorShape({1,1,1,1}), DataType::Signed32);
385
telsoa01c577f2c2018-08-31 09:22:23 +0100386 if (descriptor.m_BiasEnabled)
387 {
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000388 if(descriptor.m_ConstantWeights)
389 {
390 ARMNN_ASSERT(cLayer->m_Bias.get() != nullptr);
391 biasInfo = OverrideDataType(cLayer->m_Bias->GetTensorInfo(), GetBiasTypeFromWeightsType(dataType));
392 biasInfoPtr = &biasInfo;
393 }
394 else
395 {
396 biasInfo = OverrideDataType(layer.GetInputSlot(2).GetConnection()->GetTensorInfo(), dataType);
397 biasInfoPtr = &biasInfo;
398 }
telsoa01c577f2c2018-08-31 09:22:23 +0100399 }
400 else
401 {
402 // If biases are not enabled pass a dummy tensorinfo for the validation
403 switch(input.GetDataType())
404 {
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000405 case DataType::BFloat16:
406 {
407 biasInfoPtr = &dummyBFloat16Bias;
408 break;
409 }
telsoa01c577f2c2018-08-31 09:22:23 +0100410 case DataType::Float16:
411 {
412 biasInfoPtr = &dummyFloat16Bias;
413 break;
414 }
415 case DataType::Float32:
416 {
417 biasInfoPtr = &dummyFloat32Bias;
418 break;
419 }
Derek Lambertif90c56d2020-01-10 17:14:08 +0000420 case DataType::QAsymmU8:
Keith Davisa8565012020-02-14 12:22:40 +0000421 case DataType::QAsymmS8:
Keith Davis9d0ff742020-02-03 14:47:54 +0000422 case DataType::QSymmS8:
Derek Lambertif90c56d2020-01-10 17:14:08 +0000423 case DataType::QSymmS16:
telsoa01c577f2c2018-08-31 09:22:23 +0100424 {
425 biasInfoPtr = &dummyQA8Bias;
426 break;
427 }
428 default:
429 {
Narumol Prangnawaratac2770a2020-04-01 16:51:23 +0100430 ARMNN_ASSERT_MSG(false, "Unexpected bias type");
telsoa01c577f2c2018-08-31 09:22:23 +0100431 }
432 }
433 }
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000434 result = layerSupportObject.IsFullyConnectedSupported(
telsoa01c577f2c2018-08-31 09:22:23 +0100435 OverrideDataType(input, dataType),
436 OverrideDataType(output, dataType),
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000437 *weightsInfoPtr,
telsoa01c577f2c2018-08-31 09:22:23 +0100438 *biasInfoPtr,
439 descriptor,
David Beck33f0ae02018-10-18 15:13:56 +0100440 reason);
telsoa014fcda012018-03-09 14:13:49 +0000441 break;
442 }
narpra01b89b05f2019-01-16 09:53:09 +0000443 case LayerType::Gather:
444 {
445 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
446 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
447 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Teresa Charlin52664732020-06-29 16:27:03 +0100448 auto cLayer = PolymorphicDowncast<const GatherLayer*>(&layer);
449 const GatherDescriptor& descriptor = cLayer->GetParameters();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000450 result = layerSupportObject.IsGatherSupported(OverrideDataType(input0, dataType),
451 input1,
452 OverrideDataType(output, dataType),
453 descriptor,
454 reason);
narpra01b89b05f2019-01-16 09:53:09 +0000455 break;
456 }
telsoa014fcda012018-03-09 14:13:49 +0000457 case LayerType::Input:
458 {
459 const TensorInfo& input = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000460 result = layerSupportObject.IsInputSupported(OverrideDataType(input, dataType), reason);
telsoa014fcda012018-03-09 14:13:49 +0000461 break;
462 }
Kevin Mayce5045a2019-10-02 14:07:47 +0100463 case LayerType::InstanceNormalization:
464 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100465 auto cLayer = PolymorphicDowncast<const InstanceNormalizationLayer*>(&layer);
Kevin Mayce5045a2019-10-02 14:07:47 +0100466 const InstanceNormalizationDescriptor& descriptor = cLayer->GetParameters();
467
468 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
469 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
470
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000471 result = layerSupportObject.IsInstanceNormalizationSupported(
Kevin Mayce5045a2019-10-02 14:07:47 +0100472 OverrideDataType(input, dataType),
473 OverrideDataType(output, dataType),
474 descriptor,
475 reason);
476 break;
477 }
telsoa014fcda012018-03-09 14:13:49 +0000478 case LayerType::L2Normalization:
479 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100480 auto cLayer = PolymorphicDowncast<const L2NormalizationLayer*>(&layer);
Matteo Martincighbcd3c852018-09-28 14:14:12 +0100481 const L2NormalizationDescriptor& descriptor = cLayer->GetParameters();
482
telsoa014fcda012018-03-09 14:13:49 +0000483 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
telsoa01c577f2c2018-08-31 09:22:23 +0100484 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Matteo Martincighbcd3c852018-09-28 14:14:12 +0100485
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000486 result = layerSupportObject.IsL2NormalizationSupported(
Matteo Martincighbcd3c852018-09-28 14:14:12 +0100487 OverrideDataType(input, dataType),
488 OverrideDataType(output, dataType),
489 descriptor,
David Beck33f0ae02018-10-18 15:13:56 +0100490 reason);
telsoa01c577f2c2018-08-31 09:22:23 +0100491 break;
492 }
James Conroyaba90cd2020-11-06 16:28:18 +0000493 case LayerType::LogicalBinary:
494 {
495 auto cLayer = PolymorphicDowncast<const LogicalBinaryLayer*>(&layer);
496
497 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
498 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
499 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
500
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000501 result = layerSupportObject.IsLogicalBinarySupported(input0,
502 input1,
503 output,
504 cLayer->GetParameters(),
505 reason);
James Conroyaba90cd2020-11-06 16:28:18 +0000506 break;
507 }
Aron Virginas-Tarf982dea2019-10-11 14:07:53 +0100508 case LayerType::LogSoftmax:
509 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100510 auto cLayer = PolymorphicDowncast<const LogSoftmaxLayer*>(&layer);
Aron Virginas-Tarf982dea2019-10-11 14:07:53 +0100511
512 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
513 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
514
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000515 result = layerSupportObject.IsLogSoftmaxSupported(OverrideDataType(input, dataType),
516 OverrideDataType(output, dataType),
517 cLayer->GetParameters(),
518 reason);
Aron Virginas-Tarf982dea2019-10-11 14:07:53 +0100519 break;
520 }
telsoa01c577f2c2018-08-31 09:22:23 +0100521 case LayerType::Lstm:
522 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100523 auto cLayer = PolymorphicDowncast<const LstmLayer*>(&layer);
telsoa01c577f2c2018-08-31 09:22:23 +0100524 const LstmDescriptor& descriptor = cLayer->GetParameters();
525
526 // All inputs.
527 const TensorInfo& input = OverrideDataType(layer.GetInputSlot(0).GetConnection()->GetTensorInfo(),
528 dataType);
529 const TensorInfo& outputStateIn = OverrideDataType(layer.GetInputSlot(1).GetConnection()->GetTensorInfo(),
530 dataType);
531 const TensorInfo& cellStateIn = OverrideDataType(layer.GetInputSlot(2).GetConnection()->GetTensorInfo(),
532 dataType);
533 // All outputs
534 const TensorInfo& scratchBuffer = OverrideDataType(layer.GetOutputSlot(0).GetTensorInfo(), dataType);
535 const TensorInfo& outputStateOut = OverrideDataType(layer.GetOutputSlot(1).GetTensorInfo(), dataType);
536 const TensorInfo& cellStateOut = OverrideDataType(layer.GetOutputSlot(2).GetTensorInfo(), dataType);
537 const TensorInfo& output = OverrideDataType(layer.GetOutputSlot(3).GetTensorInfo(), dataType);
538
539 // Basic parameters
540 const TensorInfo& inputToForgetWeights
541 = OverrideDataType(cLayer->m_BasicParameters.m_InputToForgetWeights->GetTensorInfo(), dataType);
542 const TensorInfo& inputToCellWeights
543 = OverrideDataType(cLayer->m_BasicParameters.m_InputToCellWeights->GetTensorInfo(), dataType);
544 const TensorInfo& inputToOutputWeights
545 = OverrideDataType(cLayer->m_BasicParameters.m_InputToOutputWeights->GetTensorInfo(), dataType);
546 const TensorInfo& recurrentToForgetWeights
547 = OverrideDataType(cLayer->m_BasicParameters.m_RecurrentToForgetWeights->GetTensorInfo(), dataType);
548 const TensorInfo& recurrentToCellWeights
549 = OverrideDataType(cLayer->m_BasicParameters.m_RecurrentToCellWeights->GetTensorInfo(), dataType);
550 const TensorInfo& recurrentToOutputWeights
551 = OverrideDataType(cLayer->m_BasicParameters.m_RecurrentToOutputWeights->GetTensorInfo(), dataType);
552 const TensorInfo& forgetGateBias
553 = OverrideDataType(cLayer->m_BasicParameters.m_ForgetGateBias->GetTensorInfo(), dataType);
554 const TensorInfo& cellBias
555 = OverrideDataType(cLayer->m_BasicParameters.m_CellBias->GetTensorInfo(), dataType);
556 const TensorInfo& outputGateBias
557 = OverrideDataType(cLayer->m_BasicParameters.m_OutputGateBias->GetTensorInfo(), dataType);
558
Jan Eilersd01a83c2019-07-03 18:20:40 +0100559 LstmInputParamsInfo paramsInfo;
telsoa01c577f2c2018-08-31 09:22:23 +0100560
Jan Eilersd01a83c2019-07-03 18:20:40 +0100561 paramsInfo.m_InputToForgetWeights = &inputToForgetWeights;
562 paramsInfo.m_InputToCellWeights = &inputToCellWeights;
563 paramsInfo.m_InputToOutputWeights = &inputToOutputWeights;
564 paramsInfo.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
565 paramsInfo.m_RecurrentToCellWeights = &recurrentToCellWeights;
566 paramsInfo.m_RecurrentToOutputWeights = &recurrentToOutputWeights;
567 paramsInfo.m_ForgetGateBias = &forgetGateBias;
568 paramsInfo.m_CellBias = &cellBias;
569 paramsInfo.m_OutputGateBias = &outputGateBias;
570
571
572 // Optional parameters
telsoa01c577f2c2018-08-31 09:22:23 +0100573 TensorInfo optInputToInputWeights;
574 TensorInfo optRecurrentToInputWeights;
575 TensorInfo optCellToInputWeights;
576 TensorInfo optInputGateBias;
577 TensorInfo optProjectionWeights;
578 TensorInfo optProjectionBias;
579 TensorInfo optCellToForgetWeights;
580 TensorInfo optCellToOutputWeights;
Jan Eilers38e05bd2019-06-26 13:10:09 +0100581 TensorInfo optInputLayerNormWeights;
582 TensorInfo optForgetLayerNormWeights;
583 TensorInfo optCellLayerNormWeights;
584 TensorInfo optOutputLayerNormWeights;
telsoa01c577f2c2018-08-31 09:22:23 +0100585
586 if(!descriptor.m_CifgEnabled)
587 {
588 optInputToInputWeights =
589 OverrideDataType(cLayer->m_CifgParameters.m_InputToInputWeights->GetTensorInfo(), dataType);
Jan Eilersd01a83c2019-07-03 18:20:40 +0100590 paramsInfo.m_InputToInputWeights = &optInputToInputWeights;
telsoa01c577f2c2018-08-31 09:22:23 +0100591
592 optRecurrentToInputWeights =
593 OverrideDataType(cLayer->m_CifgParameters.m_RecurrentToInputWeights->GetTensorInfo(), dataType);
Jan Eilersd01a83c2019-07-03 18:20:40 +0100594 paramsInfo.m_RecurrentToInputWeights = &optRecurrentToInputWeights;
telsoa01c577f2c2018-08-31 09:22:23 +0100595 optInputGateBias =
596 OverrideDataType(cLayer->m_CifgParameters.m_InputGateBias->GetTensorInfo(), dataType);
Jan Eilersd01a83c2019-07-03 18:20:40 +0100597 paramsInfo.m_InputGateBias = &optInputGateBias;
telsoa01c577f2c2018-08-31 09:22:23 +0100598 }
599
600 if(descriptor.m_ProjectionEnabled)
601 {
602 optProjectionWeights =
603 OverrideDataType(cLayer->m_ProjectionParameters.m_ProjectionWeights->GetTensorInfo(), dataType);
Jan Eilersd01a83c2019-07-03 18:20:40 +0100604 paramsInfo.m_ProjectionWeights = &optProjectionWeights;
telsoa01c577f2c2018-08-31 09:22:23 +0100605 if (cLayer->m_ProjectionParameters.m_ProjectionBias != nullptr)
606 {
607 optProjectionBias =
608 OverrideDataType(cLayer->m_ProjectionParameters.m_ProjectionBias->GetTensorInfo(), dataType);
Jan Eilersd01a83c2019-07-03 18:20:40 +0100609 paramsInfo.m_ProjectionBias = &optProjectionBias;
telsoa01c577f2c2018-08-31 09:22:23 +0100610 }
611 }
612
613 if(descriptor.m_PeepholeEnabled)
614 {
Jan Eilerse2062cd2020-03-30 15:07:45 +0100615 if(!descriptor.m_CifgEnabled)
616 {
617 optCellToInputWeights =
618 OverrideDataType(cLayer->m_PeepholeParameters.m_CellToInputWeights->GetTensorInfo(),
619 dataType);
620 paramsInfo.m_CellToInputWeights = &optCellToInputWeights;
621 }
telsoa01c577f2c2018-08-31 09:22:23 +0100622 optCellToForgetWeights =
623 OverrideDataType(cLayer->m_PeepholeParameters.m_CellToForgetWeights->GetTensorInfo(), dataType);
Jan Eilersd01a83c2019-07-03 18:20:40 +0100624 paramsInfo.m_CellToForgetWeights = &optCellToForgetWeights;
telsoa01c577f2c2018-08-31 09:22:23 +0100625 optCellToOutputWeights =
626 OverrideDataType(cLayer->m_PeepholeParameters.m_CellToOutputWeights->GetTensorInfo(), dataType);
Jan Eilersd01a83c2019-07-03 18:20:40 +0100627 paramsInfo.m_CellToOutputWeights = &optCellToOutputWeights;
telsoa01c577f2c2018-08-31 09:22:23 +0100628 }
629
Jan Eilers38e05bd2019-06-26 13:10:09 +0100630 if(descriptor.m_LayerNormEnabled)
631 {
Ferran Balaguere30c16e2019-07-24 17:03:45 +0100632 if (!descriptor.m_CifgEnabled)
633 {
634 optInputLayerNormWeights = OverrideDataType(
635 cLayer->m_LayerNormParameters.m_InputLayerNormWeights->GetTensorInfo(), dataType);
636 paramsInfo.m_InputLayerNormWeights = &optInputLayerNormWeights;
637 }
Jan Eilers38e05bd2019-06-26 13:10:09 +0100638
639 optForgetLayerNormWeights = OverrideDataType(
640 cLayer->m_LayerNormParameters.m_ForgetLayerNormWeights->GetTensorInfo(), dataType);
Jan Eilersd01a83c2019-07-03 18:20:40 +0100641 paramsInfo.m_ForgetLayerNormWeights = &optForgetLayerNormWeights;
Jan Eilers38e05bd2019-06-26 13:10:09 +0100642
643 optCellLayerNormWeights = OverrideDataType(
644 cLayer->m_LayerNormParameters.m_CellLayerNormWeights->GetTensorInfo(), dataType);
Jan Eilersd01a83c2019-07-03 18:20:40 +0100645 paramsInfo.m_CellLayerNormWeights = &optCellLayerNormWeights;
Jan Eilers38e05bd2019-06-26 13:10:09 +0100646
647 optOutputLayerNormWeights = OverrideDataType(
648 cLayer->m_LayerNormParameters.m_OutputLayerNormWeights->GetTensorInfo(), dataType);
Jan Eilersd01a83c2019-07-03 18:20:40 +0100649 paramsInfo.m_OutputLayerNormWeights = &optOutputLayerNormWeights;
Jan Eilers38e05bd2019-06-26 13:10:09 +0100650 }
651
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000652 result = layerSupportObject.IsLstmSupported(
telsoa01c577f2c2018-08-31 09:22:23 +0100653 input,
654 outputStateIn,
655 cellStateIn,
656 scratchBuffer,
657 outputStateOut,
658 cellStateOut,
659 output,
660 descriptor,
Jan Eilersd01a83c2019-07-03 18:20:40 +0100661 paramsInfo,
662 reason);
telsoa014fcda012018-03-09 14:13:49 +0000663 break;
664 }
Nattapat Chaimanowong5a4304a2018-11-28 10:44:37 +0000665 case LayerType::Maximum:
666 {
667 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
668 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
669 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
670
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000671 result = layerSupportObject.IsMaximumSupported(OverrideDataType(input0, dataType),
672 OverrideDataType(input1, dataType),
673 OverrideDataType(output, dataType),
674 reason);
Nattapat Chaimanowong5a4304a2018-11-28 10:44:37 +0000675 break;
676 }
narpra01b89b05f2019-01-16 09:53:09 +0000677 case LayerType::MemCopy:
678 {
679 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
680 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000681
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000682 result = layerSupportObject.IsMemCopySupported(OverrideDataType(input, dataType),
683 OverrideDataType(output, dataType),
684 reason);
narpra01b89b05f2019-01-16 09:53:09 +0000685 break;
686 }
Derek Lambertif674aa02019-08-01 15:56:25 +0100687 case LayerType::MemImport:
688 {
689 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
690 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
691
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000692 result = layerSupportObject.IsMemImportSupported(OverrideDataType(input, dataType),
693 OverrideDataType(output, dataType),
694 reason);
Derek Lambertif674aa02019-08-01 15:56:25 +0100695 break;
696 }
Nattapat Chaimanowong1f886302019-04-05 13:37:19 +0100697 case LayerType::Merge:
698 {
699 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
700 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
701 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
702
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000703 result = layerSupportObject.IsMergeSupported(OverrideDataType(input0, dataType),
704 OverrideDataType(input1, dataType),
705 OverrideDataType(output, dataType),
706 reason);
Nattapat Chaimanowong1f886302019-04-05 13:37:19 +0100707 break;
708 }
Jim Flynne242f2d2019-05-22 14:24:13 +0100709 case LayerType::Concat:
telsoa014fcda012018-03-09 14:13:49 +0000710 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100711 auto cLayer = PolymorphicDowncast<const ConcatLayer*>(&layer);
telsoa014fcda012018-03-09 14:13:49 +0000712
telsoa01c577f2c2018-08-31 09:22:23 +0100713 // Get vector of all inputs.
714 auto getTensorInfo = [&dataType](const InputSlot& slot)
telsoa014fcda012018-03-09 14:13:49 +0000715 {
telsoa01c577f2c2018-08-31 09:22:23 +0100716 return OverrideDataType(slot.GetConnectedOutputSlot()->GetTensorInfo(), dataType);
telsoa014fcda012018-03-09 14:13:49 +0000717 };
Finn Williams3e54d032020-10-22 16:53:35 +0100718
719 auto beginI = MakeTransformIterator(layer.GetInputSlots().begin(), getTensorInfo);
720 auto endI = MakeTransformIterator(layer.GetInputSlots().end(), getTensorInfo);
telsoa01c577f2c2018-08-31 09:22:23 +0100721 std::vector<TensorInfo> inputs(beginI, endI);
telsoa014fcda012018-03-09 14:13:49 +0000722
telsoa01c577f2c2018-08-31 09:22:23 +0100723 auto getTensorInfoPtr = [](const TensorInfo& info)
724 {
725 return &info;
726 };
Finn Williams3e54d032020-10-22 16:53:35 +0100727
728 auto beginPtr = MakeTransformIterator(inputs.begin(), getTensorInfoPtr);
729 auto endPtr = MakeTransformIterator(inputs.end(), getTensorInfoPtr);
telsoa01c577f2c2018-08-31 09:22:23 +0100730 std::vector<const TensorInfo*> inputPtrs(beginPtr, endPtr);
telsoa014fcda012018-03-09 14:13:49 +0000731
Nikhil Raj8599a412018-11-19 14:51:07 +0000732 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
733
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000734 result = layerSupportObject.IsConcatSupported(inputPtrs, output, cLayer->GetParameters(), reason);
Jim Flynne242f2d2019-05-22 14:24:13 +0100735
736
telsoa014fcda012018-03-09 14:13:49 +0000737 break;
738 }
739 case LayerType::Multiplication:
740 {
741 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
742 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
telsoa01c577f2c2018-08-31 09:22:23 +0100743 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000744 result = layerSupportObject.IsMultiplicationSupported(
telsoa01c577f2c2018-08-31 09:22:23 +0100745 OverrideDataType(input0, dataType),
746 OverrideDataType(input1, dataType),
747 OverrideDataType(output, dataType),
David Beck33f0ae02018-10-18 15:13:56 +0100748 reason);
telsoa014fcda012018-03-09 14:13:49 +0000749 break;
750 }
751 case LayerType::Normalization:
752 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100753 auto cLayer = PolymorphicDowncast<const NormalizationLayer*>(&layer);
telsoa014fcda012018-03-09 14:13:49 +0000754 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
755 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000756 result = layerSupportObject.IsNormalizationSupported(OverrideDataType(input, dataType),
757 OverrideDataType(output, dataType),
758 cLayer->GetParameters(),
759 reason);
telsoa014fcda012018-03-09 14:13:49 +0000760 break;
761 }
762 case LayerType::Output:
763 {
764 const TensorInfo& output = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000765 result = layerSupportObject.IsOutputSupported(OverrideDataType(output, dataType), reason);
telsoa014fcda012018-03-09 14:13:49 +0000766 break;
767 }
768 case LayerType::Permute:
769 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100770 auto cLayer = PolymorphicDowncast<const PermuteLayer*>(&layer);
telsoa014fcda012018-03-09 14:13:49 +0000771 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
772 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000773 result = layerSupportObject.IsPermuteSupported(OverrideDataType(input, dataType),
774 OverrideDataType(output, dataType),
775 cLayer->GetParameters(),
776 reason);
telsoa014fcda012018-03-09 14:13:49 +0000777 break;
778 }
Mohamed Nour Abouelseoud5662c202018-09-24 13:30:09 +0100779 case LayerType::Pad:
780 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100781 auto cLayer = PolymorphicDowncast<const PadLayer*>(&layer);
Mohamed Nour Abouelseoud5662c202018-09-24 13:30:09 +0100782 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
783 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000784 result = layerSupportObject.IsPadSupported(
Mohamed Nour Abouelseoud5662c202018-09-24 13:30:09 +0100785 OverrideDataType(input, dataType),
786 OverrideDataType(output, dataType),
787 cLayer->GetParameters(),
David Beck33f0ae02018-10-18 15:13:56 +0100788 reason);
Mohamed Nour Abouelseoud5662c202018-09-24 13:30:09 +0100789 break;
790 }
telsoa014fcda012018-03-09 14:13:49 +0000791 case LayerType::Pooling2d:
792 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100793 auto cLayer = PolymorphicDowncast<const Pooling2dLayer*>(&layer);
telsoa014fcda012018-03-09 14:13:49 +0000794 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
795 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000796 result = layerSupportObject.IsPooling2dSupported(OverrideDataType(input, dataType),
797 OverrideDataType(output, dataType),
798 cLayer->GetParameters(),
799 reason);
telsoa014fcda012018-03-09 14:13:49 +0000800 break;
801 }
Matteo Martincigh49124022019-01-11 13:25:59 +0000802 case LayerType::PreCompiled:
803 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100804 auto cLayer = PolymorphicDowncast<const PreCompiledLayer*>(&layer);
Matteo Martincigh49124022019-01-11 13:25:59 +0000805 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000806 result = layerSupportObject.IsPreCompiledSupported(OverrideDataType(input, dataType),
807 cLayer->GetParameters(),
808 reason);
Matteo Martincigh49124022019-01-11 13:25:59 +0000809 break;
810 }
Derek Lambertia9cca6a2019-03-25 15:41:58 +0000811 case LayerType::Quantize:
812 {
813 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
814 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000815 result = layerSupportObject.IsQuantizeSupported(input, output, reason);
Derek Lambertia9cca6a2019-03-25 15:41:58 +0000816 break;
817 }
James Conroy586a9aa2020-03-20 08:49:33 +0000818 case LayerType::QLstm:
819 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100820 auto cLayer = PolymorphicDowncast<const QLstmLayer*>(&layer);
James Conroy586a9aa2020-03-20 08:49:33 +0000821 const QLstmDescriptor& descriptor = cLayer->GetParameters();
822
823 // Inputs
824 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
825 const TensorInfo& previousOutputIn = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
826 const TensorInfo& previousCellStateIn = layer.GetInputSlot(2).GetConnection()->GetTensorInfo();
827
828 // Outputs
829 const TensorInfo& outputStateOut = layer.GetOutputSlot(0).GetTensorInfo();
830 const TensorInfo& cellStateOut = layer.GetOutputSlot(1).GetTensorInfo();
831 const TensorInfo& output = layer.GetOutputSlot(2).GetTensorInfo();
832
833 // Lstm parameters
834 LstmInputParamsInfo paramsInfo;
835
836 // Basic parameters
837 paramsInfo.m_InputToForgetWeights = &cLayer->m_BasicParameters.m_InputToForgetWeights->GetTensorInfo();
838 paramsInfo.m_InputToCellWeights = &cLayer->m_BasicParameters.m_InputToCellWeights->GetTensorInfo();
839 paramsInfo.m_InputToOutputWeights = &cLayer->m_BasicParameters.m_InputToOutputWeights->GetTensorInfo();
840
841 paramsInfo.m_RecurrentToForgetWeights =
842 &cLayer->m_BasicParameters.m_RecurrentToForgetWeights->GetTensorInfo();
843 paramsInfo.m_RecurrentToCellWeights =
844 &cLayer->m_BasicParameters.m_RecurrentToCellWeights->GetTensorInfo();
845 paramsInfo.m_RecurrentToOutputWeights =
846 &cLayer->m_BasicParameters.m_RecurrentToOutputWeights->GetTensorInfo();
847
848 paramsInfo.m_ForgetGateBias = &cLayer->m_BasicParameters.m_ForgetGateBias->GetTensorInfo();
849 paramsInfo.m_CellBias = &cLayer->m_BasicParameters.m_CellBias->GetTensorInfo();
850 paramsInfo.m_OutputGateBias = &cLayer->m_BasicParameters.m_OutputGateBias->GetTensorInfo();
851
852 if(!descriptor.m_CifgEnabled)
853 {
854 paramsInfo.m_InputToInputWeights = &cLayer->m_CifgParameters.m_InputToInputWeights->GetTensorInfo();
855 paramsInfo.m_RecurrentToInputWeights =
856 &cLayer->m_CifgParameters.m_RecurrentToInputWeights->GetTensorInfo();
857 paramsInfo.m_InputGateBias = &cLayer->m_CifgParameters.m_InputGateBias->GetTensorInfo();
858 }
859
860 if(descriptor.m_ProjectionEnabled)
861 {
862 paramsInfo.m_ProjectionWeights = &cLayer->m_ProjectionParameters.m_ProjectionWeights->GetTensorInfo();
James Conroyed324052020-05-18 15:16:42 +0100863
864 // Projection bias is optional even if projection is enabled
865 if (cLayer->m_ProjectionParameters.m_ProjectionBias != nullptr)
866 {
867 paramsInfo.m_ProjectionBias = &cLayer->m_ProjectionParameters.m_ProjectionBias->GetTensorInfo();
868 }
James Conroy586a9aa2020-03-20 08:49:33 +0000869 }
870
871 if(descriptor.m_PeepholeEnabled)
872 {
873 if (!descriptor.m_CifgEnabled)
874 {
875 paramsInfo.m_CellToInputWeights =
876 &cLayer->m_PeepholeParameters.m_CellToInputWeights->GetTensorInfo();
877 }
878
879 paramsInfo.m_CellToForgetWeights =
880 &cLayer->m_PeepholeParameters.m_CellToForgetWeights->GetTensorInfo();
881 paramsInfo.m_CellToOutputWeights = &cLayer->m_PeepholeParameters.m_CellToOutputWeights->GetTensorInfo();
882 }
883
884 if(descriptor.m_LayerNormEnabled)
885 {
886 if (!descriptor.m_CifgEnabled)
887 {
888 paramsInfo.m_InputLayerNormWeights =
889 &cLayer->m_LayerNormParameters.m_InputLayerNormWeights->GetTensorInfo();
890 }
891
892 paramsInfo.m_ForgetLayerNormWeights =
893 &cLayer->m_LayerNormParameters.m_ForgetLayerNormWeights->GetTensorInfo();
894 paramsInfo.m_CellLayerNormWeights =
895 &cLayer->m_LayerNormParameters.m_CellLayerNormWeights->GetTensorInfo();
896 paramsInfo.m_OutputLayerNormWeights =
897 &cLayer->m_LayerNormParameters.m_OutputLayerNormWeights->GetTensorInfo();
898 }
899
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000900 result = layerSupportObject.IsQLstmSupported(input,
901 previousOutputIn,
902 previousCellStateIn,
903 outputStateOut,
904 cellStateOut,
905 output,
906 descriptor,
907 paramsInfo,
908 reason);
James Conroy586a9aa2020-03-20 08:49:33 +0000909 break;
910 }
James Conroyee18dc82019-07-17 11:27:46 +0100911 case LayerType::QuantizedLstm:
912 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100913 auto cLayer = PolymorphicDowncast<const QuantizedLstmLayer*>(&layer);
James Conroyee18dc82019-07-17 11:27:46 +0100914
915 // Inputs
Ferran Balaguer737d9ff2019-08-01 09:58:08 +0100916 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
917 const TensorInfo& previousCellStateIn = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
918 const TensorInfo& previousOutputIn = layer.GetInputSlot(2).GetConnection()->GetTensorInfo();
James Conroyee18dc82019-07-17 11:27:46 +0100919
920 // Outputs
Ferran Balaguer737d9ff2019-08-01 09:58:08 +0100921 const TensorInfo& cellStateOut = layer.GetOutputSlot(0).GetTensorInfo();
922 const TensorInfo& output = layer.GetOutputSlot(1).GetTensorInfo();
James Conroyee18dc82019-07-17 11:27:46 +0100923
924 // QuantizedLstm parameters
James Conroyee18dc82019-07-17 11:27:46 +0100925 QuantizedLstmInputParamsInfo paramsInfo;
926
Ferran Balaguer737d9ff2019-08-01 09:58:08 +0100927 paramsInfo.m_InputToInputWeights =
928 &cLayer->m_QuantizedLstmParameters.m_InputToInputWeights->GetTensorInfo();
929 paramsInfo.m_InputToForgetWeights =
930 &cLayer->m_QuantizedLstmParameters.m_InputToForgetWeights->GetTensorInfo();
931 paramsInfo.m_InputToCellWeights =
932 &cLayer->m_QuantizedLstmParameters.m_InputToCellWeights->GetTensorInfo();
933 paramsInfo.m_InputToOutputWeights =
934 &cLayer->m_QuantizedLstmParameters.m_InputToOutputWeights->GetTensorInfo();
James Conroyee18dc82019-07-17 11:27:46 +0100935
Ferran Balaguer737d9ff2019-08-01 09:58:08 +0100936 paramsInfo.m_RecurrentToInputWeights =
937 &cLayer->m_QuantizedLstmParameters.m_RecurrentToInputWeights->GetTensorInfo();
938 paramsInfo.m_RecurrentToForgetWeights =
939 &cLayer->m_QuantizedLstmParameters.m_RecurrentToForgetWeights->GetTensorInfo();
940 paramsInfo.m_RecurrentToCellWeights =
941 &cLayer->m_QuantizedLstmParameters.m_RecurrentToCellWeights->GetTensorInfo();
942 paramsInfo.m_RecurrentToOutputWeights =
943 &cLayer->m_QuantizedLstmParameters.m_RecurrentToOutputWeights->GetTensorInfo();
James Conroyee18dc82019-07-17 11:27:46 +0100944
Ferran Balaguer737d9ff2019-08-01 09:58:08 +0100945 paramsInfo.m_InputGateBias =
946 &cLayer->m_QuantizedLstmParameters.m_InputGateBias->GetTensorInfo();
947 paramsInfo.m_ForgetGateBias =
948 &cLayer->m_QuantizedLstmParameters.m_ForgetGateBias->GetTensorInfo();
949 paramsInfo.m_CellBias =
950 &cLayer->m_QuantizedLstmParameters.m_CellBias->GetTensorInfo();
951 paramsInfo.m_OutputGateBias =
952 &cLayer->m_QuantizedLstmParameters.m_OutputGateBias->GetTensorInfo();;
James Conroyee18dc82019-07-17 11:27:46 +0100953
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000954 result = layerSupportObject.IsQuantizedLstmSupported(input,
955 previousCellStateIn,
956 previousOutputIn,
957 cellStateOut,
958 output,
959 paramsInfo,
960 reason);
James Conroyee18dc82019-07-17 11:27:46 +0100961 break;
962 }
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100963 case LayerType::Division:
964 {
965 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
966 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
967 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000968 result = layerSupportObject.IsDivisionSupported(
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100969 OverrideDataType(input0, dataType),
970 OverrideDataType(input1, dataType),
971 OverrideDataType(output, dataType),
David Beck33f0ae02018-10-18 15:13:56 +0100972 reason);
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100973 break;
974 }
Finn Williams2605b232020-06-10 15:53:46 +0100975 case LayerType::Rank:
976 {
977 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
978 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000979 result = layerSupportObject.IsRankSupported(OverrideDataType(input, dataType),
980 OverrideDataType(output, dataType),
981 reason);
Finn Williams2605b232020-06-10 15:53:46 +0100982 break;
983 }
telsoa014fcda012018-03-09 14:13:49 +0000984 case LayerType::Reshape:
985 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100986 auto cLayer = PolymorphicDowncast<const ReshapeLayer*>(&layer);
telsoa014fcda012018-03-09 14:13:49 +0000987 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
Kevin Maya023c402019-12-12 17:28:05 +0000988 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000989 result = layerSupportObject.IsReshapeSupported(OverrideDataType(input, dataType),
990 OverrideDataType(output, dataType),
991 cLayer->GetParameters(),
992 reason);
telsoa014fcda012018-03-09 14:13:49 +0000993 break;
994 }
Teresa Charlina9075df2019-06-27 15:41:57 +0100995 case LayerType::Resize:
996 {
Jan Eilersbb446e52020-04-02 13:56:54 +0100997 auto cLayer = PolymorphicDowncast<const ResizeLayer*>(&layer);
Aron Virginas-Tar169d2f12019-07-01 19:01:44 +0100998 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
Teresa Charlina9075df2019-06-27 15:41:57 +0100999 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001000 result = layerSupportObject.IsResizeSupported(OverrideDataType(input, dataType),
1001 OverrideDataType(output, dataType),
1002 cLayer->GetParameters(),
1003 reason);
Teresa Charlina9075df2019-06-27 15:41:57 +01001004 break;
1005 }
Aron Virginas-Tar636ab402019-09-16 14:27:45 +01001006 case LayerType::Slice:
1007 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001008 auto cLayer = PolymorphicDowncast<const SliceLayer*>(&layer);
Aron Virginas-Tar636ab402019-09-16 14:27:45 +01001009
1010 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1011 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
1012
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001013 result = layerSupportObject.IsSliceSupported(OverrideDataType(input, dataType),
1014 OverrideDataType(output, dataType),
1015 cLayer->GetParameters(),
1016 reason);
Aron Virginas-Tar636ab402019-09-16 14:27:45 +01001017 break;
1018 }
telsoa014fcda012018-03-09 14:13:49 +00001019 case LayerType::Softmax:
1020 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001021 auto cLayer = PolymorphicDowncast<const SoftmaxLayer*>(&layer);
telsoa014fcda012018-03-09 14:13:49 +00001022 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
telsoa01c577f2c2018-08-31 09:22:23 +01001023 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001024 result = layerSupportObject.IsSoftmaxSupported(OverrideDataType(input, dataType),
1025 OverrideDataType(output, dataType),
1026 cLayer->GetParameters(),
1027 reason);
telsoa014fcda012018-03-09 14:13:49 +00001028 break;
1029 }
Nattapat Chaimanowong207ef9a2018-11-02 10:57:25 +00001030 case LayerType::SpaceToBatchNd:
1031 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001032 auto cLayer = PolymorphicDowncast<const SpaceToBatchNdLayer*>(&layer);
Nattapat Chaimanowong207ef9a2018-11-02 10:57:25 +00001033 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1034 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001035 result = layerSupportObject.IsSpaceToBatchNdSupported(OverrideDataType(input, dataType),
1036 OverrideDataType(output, dataType),
1037 cLayer->GetParameters(),
1038 reason);
Nattapat Chaimanowong207ef9a2018-11-02 10:57:25 +00001039 break;
1040 }
Aron Virginas-Tar972af152019-06-11 14:14:03 +01001041 case LayerType::SpaceToDepth:
1042 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001043 auto cLayer = PolymorphicDowncast<const SpaceToDepthLayer*>(&layer);
Aron Virginas-Tar972af152019-06-11 14:14:03 +01001044
1045 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1046 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
1047
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001048 result = layerSupportObject.IsSpaceToDepthSupported(OverrideDataType(input, dataType),
1049 OverrideDataType(output, dataType),
1050 cLayer->GetParameters(),
1051 reason);
Aron Virginas-Tar972af152019-06-11 14:14:03 +01001052 break;
1053 }
telsoa014fcda012018-03-09 14:13:49 +00001054 case LayerType::Splitter:
1055 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001056 auto cLayer = PolymorphicDowncast<const SplitterLayer*>(&layer);
telsoa014fcda012018-03-09 14:13:49 +00001057 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +01001058
1059 // Get vector of all outputs.
1060 auto getTensorInfo = [&dataType](const OutputSlot& slot)
1061 {
1062 return OverrideDataType(slot.GetTensorInfo(), dataType);
1063 };
Finn Williams3e54d032020-10-22 16:53:35 +01001064 auto beginI = MakeTransformIterator(layer.GetOutputSlots().begin(), getTensorInfo);
1065 auto endI = MakeTransformIterator(layer.GetOutputSlots().end(), getTensorInfo);
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +01001066 std::vector<TensorInfo> outputs(beginI, endI);
1067
1068 const std::vector<std::reference_wrapper<TensorInfo>> outputPtrs(outputs.begin(), outputs.end());
1069
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001070 result = layerSupportObject.IsSplitterSupported(OverrideDataType(input, dataType),
1071 outputPtrs,
1072 cLayer->GetParameters(),
1073 reason);
telsoa014fcda012018-03-09 14:13:49 +00001074 break;
1075 }
Matthew Jackson2b8c1da2019-07-04 14:59:16 +01001076 case LayerType::Stack:
1077 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001078 auto cLayer = PolymorphicDowncast<const StackLayer*>(&layer);
Matthew Jackson2b8c1da2019-07-04 14:59:16 +01001079
1080 // Get vector of all inputs.
1081 auto getTensorInfo = [&dataType](const InputSlot& slot)
1082 {
1083 return OverrideDataType(slot.GetConnectedOutputSlot()->GetTensorInfo(), dataType);
1084 };
Finn Williams3e54d032020-10-22 16:53:35 +01001085 auto beginI = MakeTransformIterator(layer.GetInputSlots().begin(), getTensorInfo);
1086 auto endI = MakeTransformIterator(layer.GetInputSlots().end(), getTensorInfo);
Matthew Jackson2b8c1da2019-07-04 14:59:16 +01001087 std::vector<TensorInfo> inputs(beginI, endI);
1088
1089 auto getTensorInfoPtr = [](const TensorInfo& info)
1090 {
1091 return &info;
1092 };
Finn Williams3e54d032020-10-22 16:53:35 +01001093 auto beginPtr = MakeTransformIterator(inputs.begin(), getTensorInfoPtr);
1094 auto endPtr = MakeTransformIterator(inputs.end(), getTensorInfoPtr);
Matthew Jackson2b8c1da2019-07-04 14:59:16 +01001095 std::vector<const TensorInfo*> inputPtrs(beginPtr, endPtr);
1096
1097 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
1098
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001099 result = layerSupportObject.IsStackSupported(inputPtrs, output, cLayer->GetParameters(), reason);
Matthew Jackson2b8c1da2019-07-04 14:59:16 +01001100
1101 break;
1102 }
Derek Lamberti013c3902019-10-21 10:46:16 +01001103 case LayerType::StandIn:
1104 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001105 auto cLayer = PolymorphicDowncast<const StandInLayer*>(&layer);
Derek Lamberti013c3902019-10-21 10:46:16 +01001106
1107 // Get vector of all inputs.
1108 auto getTensorInfoIn = [&dataType](const InputSlot& slot)
1109 {
1110 return OverrideDataType(slot.GetConnectedOutputSlot()->GetTensorInfo(), dataType);
1111 };
1112 auto getTensorInfoOut = [&dataType](const OutputSlot& slot)
1113 {
1114 return OverrideDataType(slot.GetTensorInfo(), dataType);
1115 };
Finn Williams3e54d032020-10-22 16:53:35 +01001116 auto beginI = MakeTransformIterator(layer.GetInputSlots().begin(), getTensorInfoIn);
1117 auto endI = MakeTransformIterator(layer.GetInputSlots().end(), getTensorInfoIn);
Derek Lamberti013c3902019-10-21 10:46:16 +01001118 std::vector<TensorInfo> inputs(beginI, endI);
1119
Finn Williams3e54d032020-10-22 16:53:35 +01001120 auto beginO = MakeTransformIterator(layer.GetOutputSlots().begin(), getTensorInfoOut);
1121 auto endO = MakeTransformIterator(layer.GetOutputSlots().end(), getTensorInfoOut);
Derek Lamberti013c3902019-10-21 10:46:16 +01001122 std::vector<TensorInfo> outputs(beginO, endO);
1123
1124
1125 auto getTensorInfoPtr = [](const TensorInfo& info)
1126 {
1127 return &info;
1128 };
Finn Williams3e54d032020-10-22 16:53:35 +01001129 auto beginPtrI = MakeTransformIterator(inputs.begin(), getTensorInfoPtr);
1130 auto endPtrI = MakeTransformIterator(inputs.end(), getTensorInfoPtr);
Derek Lamberti013c3902019-10-21 10:46:16 +01001131 std::vector<const TensorInfo*> inputPtrs(beginPtrI, endPtrI);
1132
Finn Williams3e54d032020-10-22 16:53:35 +01001133 auto beginPtrO = MakeTransformIterator(outputs.begin(), getTensorInfoPtr);
1134 auto endPtrO = MakeTransformIterator(outputs.end(), getTensorInfoPtr);
Derek Lamberti013c3902019-10-21 10:46:16 +01001135 std::vector<const TensorInfo*> outputPtrs(beginPtrO, endPtrO);
1136
1137
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001138 result = layerSupportObject.IsStandInSupported(inputPtrs,
1139 outputPtrs,
1140 cLayer->GetParameters(),
1141 reason);
Derek Lamberti013c3902019-10-21 10:46:16 +01001142 break;
1143 }
Conor Kennedy430b5d82018-11-14 15:28:28 +00001144 case LayerType::StridedSlice:
1145 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001146 auto cLayer = PolymorphicDowncast<const StridedSliceLayer*>(&layer);
Conor Kennedy430b5d82018-11-14 15:28:28 +00001147 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1148 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001149 result = layerSupportObject.IsStridedSliceSupported(OverrideDataType(input, dataType),
1150 OverrideDataType(output, dataType),
1151 cLayer->GetParameters(),
1152 reason);
Conor Kennedy430b5d82018-11-14 15:28:28 +00001153 break;
1154 }
David Beckc2044fe2018-09-05 15:00:38 +01001155 case LayerType::Subtraction:
1156 {
1157 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1158 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
1159 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001160 result = layerSupportObject.IsSubtractionSupported(
David Beckc2044fe2018-09-05 15:00:38 +01001161 OverrideDataType(input0, dataType),
1162 OverrideDataType(input1, dataType),
1163 OverrideDataType(output, dataType),
David Beck33f0ae02018-10-18 15:13:56 +01001164 reason);
David Beckc2044fe2018-09-05 15:00:38 +01001165 break;
1166 }
Sadik Armaganeff363d2019-04-05 15:25:46 +01001167 case LayerType::Switch:
1168 {
1169 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1170 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
1171 const TensorInfo& output0 = layer.GetOutputSlot(0).GetTensorInfo();
1172 const TensorInfo& output1 = layer.GetOutputSlot(1).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001173 result = layerSupportObject.IsSwitchSupported(OverrideDataType(input0, dataType),
1174 OverrideDataType(input1, dataType),
1175 OverrideDataType(output0, dataType),
1176 OverrideDataType(output1, dataType),
1177 reason);
Sadik Armaganeff363d2019-04-05 15:25:46 +01001178 break;
1179 }
narpra0132b90462018-09-13 11:07:48 +01001180 case LayerType::Mean:
1181 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001182 auto cLayer = PolymorphicDowncast<const MeanLayer*>(&layer);
narpra0132b90462018-09-13 11:07:48 +01001183 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1184 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001185 result = layerSupportObject.IsMeanSupported(
narpra0132b90462018-09-13 11:07:48 +01001186 OverrideDataType(input, dataType),
1187 OverrideDataType(output, dataType),
1188 cLayer->GetParameters(),
David Beck33f0ae02018-10-18 15:13:56 +01001189 reason);
narpra0132b90462018-09-13 11:07:48 +01001190 break;
1191 }
kevmay0190539692018-11-29 08:40:19 +00001192 case LayerType::Minimum:
1193 {
1194 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1195 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
1196 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001197 result = layerSupportObject.IsMinimumSupported(OverrideDataType(input0, dataType),
1198 OverrideDataType(input1, dataType),
1199 OverrideDataType(output, dataType),
1200 reason);
kevmay0190539692018-11-29 08:40:19 +00001201 break;
1202 }
Matteo Martincigh0e406ee2019-06-12 15:42:18 +01001203 case LayerType::Prelu:
1204 {
1205 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1206 const TensorInfo& alpha = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
1207 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001208 result = layerSupportObject.IsPreluSupported(OverrideDataType(input, dataType),
1209 OverrideDataType(alpha, dataType),
1210 OverrideDataType(output, dataType),
1211 reason);
Matteo Martincigh0e406ee2019-06-12 15:42:18 +01001212 break;
1213 }
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001214 case LayerType::Transpose:
1215 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001216 auto cLayer = PolymorphicDowncast<const TransposeLayer*>(&layer);
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001217 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1218 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001219 result = layerSupportObject.IsTransposeSupported(OverrideDataType(input, dataType),
1220 OverrideDataType(output, dataType),
1221 cLayer->GetParameters(),
1222 reason);
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001223 break;
1224 }
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001225 case LayerType::TransposeConvolution2d:
1226 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001227 auto cLayer = PolymorphicDowncast<const TransposeConvolution2dLayer*>(&layer);
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001228
1229 const TensorInfo input = OverrideDataType(layer.GetInputSlot(0).GetConnection()->GetTensorInfo(),
1230 dataType);
1231 const TensorInfo output = OverrideDataType(layer.GetOutputSlot(0).GetTensorInfo(), dataType);
1232
1233 const TransposeConvolution2dDescriptor& descriptor = cLayer->GetParameters();
1234
1235 Optional<TensorInfo> biases;
1236 if (descriptor.m_BiasEnabled)
1237 {
Narumol Prangnawaratac2770a2020-04-01 16:51:23 +01001238 ARMNN_ASSERT(cLayer->m_Bias.get() != nullptr);
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001239 biases = OverrideDataType(cLayer->m_Bias->GetTensorInfo(),
1240 GetBiasTypeFromWeightsType(dataType));
1241 }
1242
Narumol Prangnawaratac2770a2020-04-01 16:51:23 +01001243 ARMNN_ASSERT(cLayer->m_Weight.get() != nullptr);
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001244 const TensorInfo weights = OverrideDataType(cLayer->m_Weight->GetTensorInfo(), dataType);
1245
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001246 result = layerSupportObject.IsTransposeConvolution2dSupported(input,
1247 output,
1248 descriptor,
1249 weights,
1250 biases,
1251 reason);
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001252
1253 break;
1254 }
Sadik Armagan0c3ea5b2021-02-03 09:29:30 +00001255 case LayerType::Reduce:
1256 {
1257 auto cLayer = PolymorphicDowncast<const ReduceLayer*>(&layer);
1258 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1259 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
1260
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001261 result = layerSupportObject.IsReduceSupported(OverrideDataType(input, dataType),
1262 OverrideDataType(output, dataType),
1263 cLayer->GetParameters(),
1264 reason);
Sadik Armagan0c3ea5b2021-02-03 09:29:30 +00001265 break;
1266 }
telsoa014fcda012018-03-09 14:13:49 +00001267 default:
1268 {
Narumol Prangnawaratac2770a2020-04-01 16:51:23 +01001269 ARMNN_ASSERT_MSG(false, "WorkloadFactory did not recognise type of layer.");
David Beck33f0ae02018-10-18 15:13:56 +01001270 reason.value() = "Unrecognised layer type";
telsoa014fcda012018-03-09 14:13:49 +00001271 result = false;
1272 break;
1273 }
1274 }
telsoa014fcda012018-03-09 14:13:49 +00001275 return result;
1276}
1277
Sadik Armagan045f6be2020-09-10 13:37:32 +01001278bool IWorkloadFactory::IsLayerSupported(const BackendId& backendId,
1279 const IConnectableLayer& connectableLayer,
1280 Optional<DataType> dataType,
1281 std::string& outReasonIfUnsupported)
1282{
1283 return IsLayerConfigurationSupported(backendId, connectableLayer, dataType, outReasonIfUnsupported);
1284}
1285
David Beckdcb751f2018-10-03 11:42:42 +01001286bool IWorkloadFactory::IsLayerSupported(const IConnectableLayer& connectableLayer,
David Beck29c75de2018-10-23 13:35:58 +01001287 Optional<DataType> dataType,
telsoa01c577f2c2018-08-31 09:22:23 +01001288 std::string& outReasonIfUnsupported)
telsoa014fcda012018-03-09 14:13:49 +00001289{
Jan Eilersbb446e52020-04-02 13:56:54 +01001290 auto layer = PolymorphicDowncast<const Layer*>(&connectableLayer);
Sadik Armagan045f6be2020-09-10 13:37:32 +01001291 return IsLayerConfigurationSupported(layer->GetBackendId(), connectableLayer, dataType, outReasonIfUnsupported);
1292}
1293
1294// TODO merge with defaulted modelOptions above
1295bool IWorkloadFactory::IsLayerSupported(const IConnectableLayer& connectableLayer,
1296 Optional<DataType> dataType,
1297 std::string& outReasonIfUnsupported,
1298 const ModelOptions& modelOptions)
1299{
1300 auto layer = PolymorphicDowncast<const Layer*>(&connectableLayer);
1301 return IsLayerConfigurationSupported(layer->GetBackendId(),
1302 connectableLayer,
1303 dataType,
1304 outReasonIfUnsupported,
1305 modelOptions);
telsoa014fcda012018-03-09 14:13:49 +00001306}
1307
Sadik Armagan04a72972020-09-14 15:44:18 +01001308bool IWorkloadFactory::IsLayerSupported(const BackendId& backendId,
1309 const IConnectableLayer& connectableLayer,
1310 Optional<DataType> dataType,
1311 std::string& outReasonIfUnsupported,
1312 const ModelOptions& modelOptions)
1313{
1314 return IsLayerConfigurationSupported(backendId,
1315 connectableLayer,
1316 dataType,
1317 outReasonIfUnsupported,
1318 modelOptions);
1319}
1320
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001321// Default Implementations
Derek Lamberti901ea112019-12-10 22:07:09 +00001322std::unique_ptr<IWorkload> IWorkloadFactory::CreateAbs(const AbsQueueDescriptor& /*descriptor*/,
1323 const WorkloadInfo& /*info*/) const
Kevin May868eb142019-09-04 17:29:31 +01001324{
1325 return std::unique_ptr<IWorkload>();
1326}
1327
Derek Lamberti901ea112019-12-10 22:07:09 +00001328std::unique_ptr<IWorkload> IWorkloadFactory::CreateActivation(const ActivationQueueDescriptor& /*descriptor*/,
1329 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001330{
1331 return std::unique_ptr<IWorkload>();
1332}
1333
Derek Lamberti901ea112019-12-10 22:07:09 +00001334std::unique_ptr<IWorkload> IWorkloadFactory::CreateAddition(const AdditionQueueDescriptor& /*descriptor*/,
1335 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001336{
1337 return std::unique_ptr<IWorkload>();
1338}
1339
Derek Lamberti901ea112019-12-10 22:07:09 +00001340std::unique_ptr<IWorkload> IWorkloadFactory::CreateArgMinMax(const ArgMinMaxQueueDescriptor& /*descriptor*/,
1341 const WorkloadInfo& /*info*/) const
Nikhil Rajee391d52019-09-05 17:50:44 +01001342{
1343 return std::unique_ptr<IWorkload>();
1344}
1345
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001346std::unique_ptr<IWorkload> IWorkloadFactory::CreateBatchNormalization(
Derek Lamberti901ea112019-12-10 22:07:09 +00001347 const BatchNormalizationQueueDescriptor& /*descriptor*/, const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001348{
1349 return std::unique_ptr<IWorkload>();
1350}
1351
Derek Lamberti901ea112019-12-10 22:07:09 +00001352std::unique_ptr<IWorkload> IWorkloadFactory::CreateBatchToSpaceNd(const BatchToSpaceNdQueueDescriptor& /*desc*/,
1353 const WorkloadInfo& /*Info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001354{
1355 return std::unique_ptr<IWorkload>();
1356}
1357
mathad01b392e982021-04-07 12:07:30 +01001358std::unique_ptr<IWorkload> IWorkloadFactory::CreateCast(const CastQueueDescriptor& /*descriptor*/,
1359 const WorkloadInfo& /*info*/) const
1360{
1361 return std::unique_ptr<IWorkload>();
1362}
1363
Derek Lamberti901ea112019-12-10 22:07:09 +00001364std::unique_ptr<IWorkload> IWorkloadFactory::CreateComparison(const ComparisonQueueDescriptor& /*descriptor*/,
1365 const WorkloadInfo& /*info*/) const
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +01001366{
1367 return std::unique_ptr<IWorkload>();
1368}
1369
Derek Lamberti901ea112019-12-10 22:07:09 +00001370std::unique_ptr<IWorkload> IWorkloadFactory::CreateConcat(const ConcatQueueDescriptor& /*descriptor*/,
1371 const WorkloadInfo& /*info*/) const
Jim Flynn4ed6c832019-05-20 11:02:46 +01001372{
1373 return std::unique_ptr<IWorkload>();
1374}
1375
Derek Lamberti901ea112019-12-10 22:07:09 +00001376std::unique_ptr<IWorkload> IWorkloadFactory::CreateConstant(const ConstantQueueDescriptor& /*descriptor*/,
1377 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001378{
1379 return std::unique_ptr<IWorkload>();
1380}
1381
Narumol Prangnawarat7ddbbae2020-03-13 10:26:05 +00001382std::unique_ptr<IWorkload> IWorkloadFactory::CreateConvertBf16ToFp32(const ConvertBf16ToFp32QueueDescriptor& /*desc*/,
1383 const WorkloadInfo& /*info*/) const
1384{
1385 return std::unique_ptr<IWorkload>();
1386}
1387
Derek Lamberti901ea112019-12-10 22:07:09 +00001388std::unique_ptr<IWorkload> IWorkloadFactory::CreateConvertFp16ToFp32(const ConvertFp16ToFp32QueueDescriptor& /*desc*/,
1389 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001390{
1391 return std::unique_ptr<IWorkload>();
1392}
1393
Narumol Prangnawaratea54a012020-03-16 16:36:10 +00001394std::unique_ptr<IWorkload> IWorkloadFactory::CreateConvertFp32ToBf16(const ConvertFp32ToBf16QueueDescriptor& /*desc*/,
1395 const WorkloadInfo& /*info*/) const
1396{
1397 return std::unique_ptr<IWorkload>();
1398}
1399
Derek Lamberti901ea112019-12-10 22:07:09 +00001400std::unique_ptr<IWorkload> IWorkloadFactory::CreateConvertFp32ToFp16(const ConvertFp32ToFp16QueueDescriptor& /*desc*/,
1401 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001402{
1403 return std::unique_ptr<IWorkload>();
1404}
1405
Derek Lamberti901ea112019-12-10 22:07:09 +00001406std::unique_ptr<IWorkload> IWorkloadFactory::CreateConvolution2d(const Convolution2dQueueDescriptor& /*descriptor*/,
1407 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001408{
1409 return std::unique_ptr<IWorkload>();
1410}
1411
Derek Lamberti901ea112019-12-10 22:07:09 +00001412std::unique_ptr<IWorkload> IWorkloadFactory::CreateDebug(const DebugQueueDescriptor& /*descriptor*/,
1413 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001414{
1415 return std::unique_ptr<IWorkload>();
1416}
1417
Derek Lamberti901ea112019-12-10 22:07:09 +00001418std::unique_ptr<IWorkload> IWorkloadFactory::CreateDepthToSpace(const DepthToSpaceQueueDescriptor& /*descriptor*/,
1419 const WorkloadInfo& /*info*/) const
Aron Virginas-Tardd6247f2019-09-19 14:31:17 +01001420{
1421 return std::unique_ptr<IWorkload>();
1422}
1423
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001424std::unique_ptr<IWorkload> IWorkloadFactory::CreateDepthwiseConvolution2d(
Derek Lamberti901ea112019-12-10 22:07:09 +00001425 const DepthwiseConvolution2dQueueDescriptor& /*descriptor*/, const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001426{
1427 return std::unique_ptr<IWorkload>();
1428}
1429
Nattapat Chaimanowonge4294fd2019-03-28 09:56:53 +00001430std::unique_ptr<IWorkload> IWorkloadFactory::CreateDequantize(
Derek Lamberti901ea112019-12-10 22:07:09 +00001431 const DequantizeQueueDescriptor& /*descriptor*/, const WorkloadInfo& /*info*/) const
Nattapat Chaimanowonge4294fd2019-03-28 09:56:53 +00001432{
1433 return std::unique_ptr<IWorkload>();
1434}
1435
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001436std::unique_ptr<IWorkload> IWorkloadFactory::CreateDetectionPostProcess(
Derek Lamberti901ea112019-12-10 22:07:09 +00001437 const DetectionPostProcessQueueDescriptor& /*descriptor*/, const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001438{
1439 return std::unique_ptr<IWorkload>();
1440}
1441
Derek Lamberti901ea112019-12-10 22:07:09 +00001442std::unique_ptr<IWorkload> IWorkloadFactory::CreateDivision(const DivisionQueueDescriptor& /*descriptor*/,
1443 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001444{
1445 return std::unique_ptr<IWorkload>();
1446}
1447
josh minor4a3c6102020-01-06 16:40:46 -06001448std::unique_ptr<IWorkload> IWorkloadFactory::CreateElementwiseUnary(const ElementwiseUnaryQueueDescriptor& /*desc*/,
1449 const WorkloadInfo& /*info*/) const
1450{
1451 return std::unique_ptr<IWorkload>();
1452}
1453
Derek Lamberti901ea112019-12-10 22:07:09 +00001454std::unique_ptr<IWorkload> IWorkloadFactory::CreateEqual(const EqualQueueDescriptor& /*descriptor*/,
1455 const WorkloadInfo& /*Info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001456{
1457 return std::unique_ptr<IWorkload>();
1458}
1459
Derek Lamberti901ea112019-12-10 22:07:09 +00001460std::unique_ptr<IWorkload> IWorkloadFactory::CreateFakeQuantization(const FakeQuantizationQueueDescriptor& /*desc*/,
1461 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001462{
1463 return std::unique_ptr<IWorkload>();
1464}
1465
Ryan OSheaec6c6802020-06-05 17:17:06 +01001466std::unique_ptr<IWorkload> IWorkloadFactory::CreateFill(const FillQueueDescriptor& /*descriptor*/,
1467 const WorkloadInfo& /*info*/) const
1468{
1469 return std::unique_ptr<IWorkload>();
1470}
1471
Derek Lamberti901ea112019-12-10 22:07:09 +00001472std::unique_ptr<IWorkload> IWorkloadFactory::CreateFloor(const FloorQueueDescriptor& /*descriptor*/,
1473 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001474{
1475 return std::unique_ptr<IWorkload>();
1476}
1477
Derek Lamberti901ea112019-12-10 22:07:09 +00001478std::unique_ptr<IWorkload> IWorkloadFactory::CreateFullyConnected(const FullyConnectedQueueDescriptor& /*descriptor*/,
1479 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001480{
1481 return std::unique_ptr<IWorkload>();
1482}
1483
Derek Lamberti901ea112019-12-10 22:07:09 +00001484std::unique_ptr<IWorkload> IWorkloadFactory::CreateGather(const GatherQueueDescriptor& /*descriptor*/,
1485 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001486{
1487 return std::unique_ptr<IWorkload>();
1488}
1489
Derek Lamberti901ea112019-12-10 22:07:09 +00001490std::unique_ptr<IWorkload> IWorkloadFactory::CreateGreater(const GreaterQueueDescriptor& /*descriptor*/,
1491 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001492{
1493 return std::unique_ptr<IWorkload>();
1494}
1495
Kevin Mayce5045a2019-10-02 14:07:47 +01001496std::unique_ptr<IWorkload> IWorkloadFactory::CreateInstanceNormalization(
Derek Lamberti901ea112019-12-10 22:07:09 +00001497 const InstanceNormalizationQueueDescriptor& /*descriptor*/,
1498 const WorkloadInfo& /*info*/) const
Kevin Mayce5045a2019-10-02 14:07:47 +01001499{
1500 return std::unique_ptr<IWorkload>();
1501}
1502
Derek Lamberti901ea112019-12-10 22:07:09 +00001503std::unique_ptr<IWorkload> IWorkloadFactory::CreateL2Normalization(const L2NormalizationQueueDescriptor& /*desc*/,
1504 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001505{
1506 return std::unique_ptr<IWorkload>();
1507}
1508
James Conroyaba90cd2020-11-06 16:28:18 +00001509std::unique_ptr<IWorkload> IWorkloadFactory::CreateLogicalBinary(const LogicalBinaryQueueDescriptor& /*desc*/,
1510 const WorkloadInfo& /*info*/) const
1511{
1512 return std::unique_ptr<IWorkload>();
1513}
1514
1515std::unique_ptr<IWorkload> IWorkloadFactory::CreateLogicalUnary(const ElementwiseUnaryQueueDescriptor& /*desc*/,
1516 const WorkloadInfo& /*info*/) const
1517{
1518 return std::unique_ptr<IWorkload>();
1519}
1520
Derek Lamberti901ea112019-12-10 22:07:09 +00001521std::unique_ptr<IWorkload> IWorkloadFactory::CreateLogSoftmax(const LogSoftmaxQueueDescriptor& /*descriptor*/,
1522 const WorkloadInfo& /*info*/) const
Aron Virginas-Tarf982dea2019-10-11 14:07:53 +01001523{
1524 return std::unique_ptr<IWorkload>();
1525}
1526
Derek Lamberti901ea112019-12-10 22:07:09 +00001527std::unique_ptr<IWorkload> IWorkloadFactory::CreateLstm(const LstmQueueDescriptor& /*descriptor*/,
1528 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001529{
1530 return std::unique_ptr<IWorkload>();
1531}
1532
Derek Lamberti901ea112019-12-10 22:07:09 +00001533std::unique_ptr<IWorkload> IWorkloadFactory::CreateMaximum(const MaximumQueueDescriptor& /*descriptor*/,
1534 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001535{
1536 return std::unique_ptr<IWorkload>();
1537}
1538
Derek Lamberti901ea112019-12-10 22:07:09 +00001539std::unique_ptr<IWorkload> IWorkloadFactory::CreateMean(const MeanQueueDescriptor& /*descriptor*/,
1540 const WorkloadInfo& /*Info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001541{
1542 return std::unique_ptr<IWorkload>();
1543}
1544
Derek Lamberti901ea112019-12-10 22:07:09 +00001545std::unique_ptr<IWorkload> IWorkloadFactory::CreateMemCopy(const MemCopyQueueDescriptor& /*descriptor*/,
1546 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001547{
1548 return std::unique_ptr<IWorkload>();
1549}
1550
Derek Lamberti901ea112019-12-10 22:07:09 +00001551std::unique_ptr<IWorkload> IWorkloadFactory::CreateMemImport(const MemImportQueueDescriptor& /*descriptor*/,
1552 const WorkloadInfo& /*info*/) const
Derek Lambertif674aa02019-08-01 15:56:25 +01001553{
1554 return std::unique_ptr<IWorkload>();
1555}
1556
Derek Lamberti901ea112019-12-10 22:07:09 +00001557std::unique_ptr<IWorkload> IWorkloadFactory::CreateMerge(const MergeQueueDescriptor& /*descriptor*/,
1558 const WorkloadInfo& /*info*/) const
Nattapat Chaimanowong1f886302019-04-05 13:37:19 +01001559{
1560 return std::unique_ptr<IWorkload>();
1561}
1562
Derek Lamberti901ea112019-12-10 22:07:09 +00001563std::unique_ptr<IWorkload> IWorkloadFactory::CreateMerger(const MergerQueueDescriptor& /*descriptor*/,
1564 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001565{
1566 return std::unique_ptr<IWorkload>();
1567}
1568
Derek Lamberti901ea112019-12-10 22:07:09 +00001569std::unique_ptr<IWorkload> IWorkloadFactory::CreateMinimum(const MinimumQueueDescriptor& /*descriptor*/,
1570 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001571{
1572 return std::unique_ptr<IWorkload>();
1573}
1574
Derek Lamberti901ea112019-12-10 22:07:09 +00001575std::unique_ptr<IWorkload> IWorkloadFactory::CreateMultiplication(const MultiplicationQueueDescriptor& /*descriptor*/,
1576 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001577{
1578 return std::unique_ptr<IWorkload>();
1579}
1580
Derek Lamberti901ea112019-12-10 22:07:09 +00001581std::unique_ptr<IWorkload> IWorkloadFactory::CreateNormalization(const NormalizationQueueDescriptor& /*descriptor*/,
1582 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001583{
1584 return std::unique_ptr<IWorkload>();
1585}
1586
Derek Lamberti901ea112019-12-10 22:07:09 +00001587std::unique_ptr<IWorkload> IWorkloadFactory::CreateOutput(const OutputQueueDescriptor& /*descriptor*/,
1588 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001589{
1590 return std::unique_ptr<IWorkload>();
1591}
1592
Derek Lamberti901ea112019-12-10 22:07:09 +00001593std::unique_ptr<IWorkload> IWorkloadFactory::CreatePad(const PadQueueDescriptor& /*descriptor*/,
1594 const WorkloadInfo& /*Info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001595{
1596 return std::unique_ptr<IWorkload>();
1597}
1598
Derek Lamberti901ea112019-12-10 22:07:09 +00001599std::unique_ptr<IWorkload> IWorkloadFactory::CreatePermute(const PermuteQueueDescriptor& /*descriptor*/,
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001600 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001601{
1602 return std::unique_ptr<IWorkload>();
1603}
1604
Derek Lamberti901ea112019-12-10 22:07:09 +00001605std::unique_ptr<IWorkload> IWorkloadFactory::CreatePooling2d(const Pooling2dQueueDescriptor& /*descriptor*/,
1606 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001607{
1608 return std::unique_ptr<IWorkload>();
1609}
1610
Derek Lamberti901ea112019-12-10 22:07:09 +00001611std::unique_ptr<IWorkload> IWorkloadFactory::CreatePreCompiled(const PreCompiledQueueDescriptor& /*descriptor*/,
1612 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001613{
1614 return std::unique_ptr<IWorkload>();
1615}
1616
Derek Lamberti901ea112019-12-10 22:07:09 +00001617std::unique_ptr<IWorkload> IWorkloadFactory::CreatePrelu(const PreluQueueDescriptor &/*descriptor*/,
1618 const WorkloadInfo &/*info*/) const
Matteo Martincigh0e406ee2019-06-12 15:42:18 +01001619{
1620 return std::unique_ptr<IWorkload>();
1621}
1622
Derek Lamberti901ea112019-12-10 22:07:09 +00001623std::unique_ptr<IWorkload> IWorkloadFactory::CreateQuantize(const QuantizeQueueDescriptor& /*descriptor*/,
1624 const WorkloadInfo& /*Info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001625{
1626 return std::unique_ptr<IWorkload>();
1627}
1628
James Conroy586a9aa2020-03-20 08:49:33 +00001629std::unique_ptr<IWorkload> IWorkloadFactory::CreateQLstm(const QLstmQueueDescriptor& /*descriptor*/,
1630 const WorkloadInfo& /*info*/) const
1631{
1632 return std::unique_ptr<IWorkload>();
1633}
1634
Derek Lamberti901ea112019-12-10 22:07:09 +00001635std::unique_ptr<IWorkload> IWorkloadFactory::CreateQuantizedLstm(const QuantizedLstmQueueDescriptor& /*descriptor*/,
1636 const WorkloadInfo& /*info*/) const
James Conroyee18dc82019-07-17 11:27:46 +01001637{
1638 return std::unique_ptr<IWorkload>();
1639}
Finn Williams2605b232020-06-10 15:53:46 +01001640std::unique_ptr<IWorkload> IWorkloadFactory::CreateRank(const RankQueueDescriptor& /*descriptor*/,
1641 const WorkloadInfo& /*info*/) const
1642{
1643 return std::unique_ptr<IWorkload>();
1644}
James Conroyee18dc82019-07-17 11:27:46 +01001645
Sadik Armagan0c3ea5b2021-02-03 09:29:30 +00001646std::unique_ptr<IWorkload> IWorkloadFactory::CreateReduce(const ReduceQueueDescriptor& /*descriptor*/,
1647 const WorkloadInfo& /*info*/) const
1648{
1649 return std::unique_ptr<IWorkload>();
1650}
1651
Derek Lamberti901ea112019-12-10 22:07:09 +00001652std::unique_ptr<IWorkload> IWorkloadFactory::CreateReshape(const ReshapeQueueDescriptor& /*descriptor*/,
1653 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001654{
1655 return std::unique_ptr<IWorkload>();
1656}
1657
Derek Lamberti901ea112019-12-10 22:07:09 +00001658std::unique_ptr<IWorkload> IWorkloadFactory::CreateResizeBilinear(const ResizeBilinearQueueDescriptor& /*descriptor*/,
1659 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001660{
1661 return std::unique_ptr<IWorkload>();
1662}
1663
Derek Lamberti901ea112019-12-10 22:07:09 +00001664std::unique_ptr<IWorkload> IWorkloadFactory::CreateResize(const ResizeQueueDescriptor& /*descriptor*/,
1665 const WorkloadInfo& /*info*/) const
Teresa Charlina9075df2019-06-27 15:41:57 +01001666{
1667 return std::unique_ptr<IWorkload>();
1668}
1669
Derek Lamberti901ea112019-12-10 22:07:09 +00001670std::unique_ptr<IWorkload> IWorkloadFactory::CreateRsqrt(const RsqrtQueueDescriptor& /*descriptor*/,
1671 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001672{
1673 return std::unique_ptr<IWorkload>();
1674}
1675
Derek Lamberti901ea112019-12-10 22:07:09 +00001676std::unique_ptr<IWorkload> IWorkloadFactory::CreateSlice(const SliceQueueDescriptor& /*descriptor*/,
1677 const WorkloadInfo& /*info*/) const
1678{
1679 return std::unique_ptr<IWorkload>();
1680}
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001681
Derek Lamberti901ea112019-12-10 22:07:09 +00001682std::unique_ptr<IWorkload> IWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& /*descriptor*/,
1683 const WorkloadInfo& /*info*/) const
Aron Virginas-Tar636ab402019-09-16 14:27:45 +01001684{
1685 return std::unique_ptr<IWorkload>();
1686}
1687
Derek Lamberti901ea112019-12-10 22:07:09 +00001688std::unique_ptr<IWorkload> IWorkloadFactory::CreateSplitter(const SplitterQueueDescriptor& /*descriptor*/,
1689 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001690{
1691 return std::unique_ptr<IWorkload>();
1692}
1693
Derek Lamberti901ea112019-12-10 22:07:09 +00001694std::unique_ptr<IWorkload> IWorkloadFactory::CreateSpaceToBatchNd(const SpaceToBatchNdQueueDescriptor& /*descriptor*/,
1695 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001696{
1697 return std::unique_ptr<IWorkload>();
1698}
1699
Derek Lamberti901ea112019-12-10 22:07:09 +00001700std::unique_ptr<IWorkload> IWorkloadFactory::CreateSpaceToDepth(const SpaceToDepthQueueDescriptor& /*descriptor*/,
1701 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001702{
1703 return std::unique_ptr<IWorkload>();
1704}
1705
Derek Lamberti901ea112019-12-10 22:07:09 +00001706std::unique_ptr<IWorkload> IWorkloadFactory::CreateStack(const StackQueueDescriptor& /*descriptor*/,
1707 const WorkloadInfo& /*info*/) const
Aron Virginas-Tar972af152019-06-11 14:14:03 +01001708{
1709 return std::unique_ptr<IWorkload>();
1710}
1711
Derek Lamberti901ea112019-12-10 22:07:09 +00001712std::unique_ptr<IWorkload> IWorkloadFactory::CreateStridedSlice(const StridedSliceQueueDescriptor& /*descriptor*/,
1713 const WorkloadInfo& /*info*/) const
Matthew Jackson2b8c1da2019-07-04 14:59:16 +01001714{
1715 return std::unique_ptr<IWorkload>();
1716}
1717
Derek Lamberti901ea112019-12-10 22:07:09 +00001718std::unique_ptr<IWorkload> IWorkloadFactory::CreateSubtraction(const SubtractionQueueDescriptor& /*descriptor*/,
1719 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001720{
1721 return std::unique_ptr<IWorkload>();
1722}
1723
Derek Lamberti901ea112019-12-10 22:07:09 +00001724std::unique_ptr<IWorkload> IWorkloadFactory::CreateSwitch(const SwitchQueueDescriptor& /*descriptor*/,
1725 const WorkloadInfo& /*info*/) const
Sadik Armaganeff363d2019-04-05 15:25:46 +01001726{
1727 return std::unique_ptr<IWorkload>();
1728}
1729
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001730std::unique_ptr<IWorkload> IWorkloadFactory::CreateTranspose(const TransposeQueueDescriptor& /*descriptor*/,
1731 const WorkloadInfo& /*info*/) const
1732{
1733 return std::unique_ptr<IWorkload>();
1734}
1735
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001736std::unique_ptr<IWorkload> IWorkloadFactory::CreateTransposeConvolution2d(
Derek Lamberti901ea112019-12-10 22:07:09 +00001737 const TransposeConvolution2dQueueDescriptor& /*descriptor*/,
1738 const WorkloadInfo& /*info*/) const
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001739{
1740 return std::unique_ptr<IWorkload>();
surmeh013537c2c2018-05-18 16:31:43 +01001741}
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001742
Aron Virginas-Tar636ab402019-09-16 14:27:45 +01001743} // namepsace armnn