blob: 61ad20995b97056de1bd600e63891b19b35638cb [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 }
Keith Davis3ae3f972021-05-21 16:33:48 +01001006 case LayerType::Shape:
1007 {
1008 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1009 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
1010
1011 result = layerSupportObject.IsShapeSupported(OverrideDataType(input, dataType),
1012 OverrideDataType(output, dataType),
1013 reason);
1014 break;
1015 }
Aron Virginas-Tar636ab402019-09-16 14:27:45 +01001016 case LayerType::Slice:
1017 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001018 auto cLayer = PolymorphicDowncast<const SliceLayer*>(&layer);
Aron Virginas-Tar636ab402019-09-16 14:27:45 +01001019
1020 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1021 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
1022
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001023 result = layerSupportObject.IsSliceSupported(OverrideDataType(input, dataType),
1024 OverrideDataType(output, dataType),
1025 cLayer->GetParameters(),
1026 reason);
Aron Virginas-Tar636ab402019-09-16 14:27:45 +01001027 break;
1028 }
telsoa014fcda012018-03-09 14:13:49 +00001029 case LayerType::Softmax:
1030 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001031 auto cLayer = PolymorphicDowncast<const SoftmaxLayer*>(&layer);
telsoa014fcda012018-03-09 14:13:49 +00001032 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
telsoa01c577f2c2018-08-31 09:22:23 +01001033 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001034 result = layerSupportObject.IsSoftmaxSupported(OverrideDataType(input, dataType),
1035 OverrideDataType(output, dataType),
1036 cLayer->GetParameters(),
1037 reason);
telsoa014fcda012018-03-09 14:13:49 +00001038 break;
1039 }
Nattapat Chaimanowong207ef9a2018-11-02 10:57:25 +00001040 case LayerType::SpaceToBatchNd:
1041 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001042 auto cLayer = PolymorphicDowncast<const SpaceToBatchNdLayer*>(&layer);
Nattapat Chaimanowong207ef9a2018-11-02 10:57:25 +00001043 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1044 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001045 result = layerSupportObject.IsSpaceToBatchNdSupported(OverrideDataType(input, dataType),
1046 OverrideDataType(output, dataType),
1047 cLayer->GetParameters(),
1048 reason);
Nattapat Chaimanowong207ef9a2018-11-02 10:57:25 +00001049 break;
1050 }
Aron Virginas-Tar972af152019-06-11 14:14:03 +01001051 case LayerType::SpaceToDepth:
1052 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001053 auto cLayer = PolymorphicDowncast<const SpaceToDepthLayer*>(&layer);
Aron Virginas-Tar972af152019-06-11 14:14:03 +01001054
1055 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1056 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
1057
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001058 result = layerSupportObject.IsSpaceToDepthSupported(OverrideDataType(input, dataType),
1059 OverrideDataType(output, dataType),
1060 cLayer->GetParameters(),
1061 reason);
Aron Virginas-Tar972af152019-06-11 14:14:03 +01001062 break;
1063 }
telsoa014fcda012018-03-09 14:13:49 +00001064 case LayerType::Splitter:
1065 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001066 auto cLayer = PolymorphicDowncast<const SplitterLayer*>(&layer);
telsoa014fcda012018-03-09 14:13:49 +00001067 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +01001068
1069 // Get vector of all outputs.
1070 auto getTensorInfo = [&dataType](const OutputSlot& slot)
1071 {
1072 return OverrideDataType(slot.GetTensorInfo(), dataType);
1073 };
Finn Williams3e54d032020-10-22 16:53:35 +01001074 auto beginI = MakeTransformIterator(layer.GetOutputSlots().begin(), getTensorInfo);
1075 auto endI = MakeTransformIterator(layer.GetOutputSlots().end(), getTensorInfo);
Narumol Prangnawarat15eb5832019-05-20 15:31:05 +01001076 std::vector<TensorInfo> outputs(beginI, endI);
1077
1078 const std::vector<std::reference_wrapper<TensorInfo>> outputPtrs(outputs.begin(), outputs.end());
1079
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001080 result = layerSupportObject.IsSplitterSupported(OverrideDataType(input, dataType),
1081 outputPtrs,
1082 cLayer->GetParameters(),
1083 reason);
telsoa014fcda012018-03-09 14:13:49 +00001084 break;
1085 }
Matthew Jackson2b8c1da2019-07-04 14:59:16 +01001086 case LayerType::Stack:
1087 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001088 auto cLayer = PolymorphicDowncast<const StackLayer*>(&layer);
Matthew Jackson2b8c1da2019-07-04 14:59:16 +01001089
1090 // Get vector of all inputs.
1091 auto getTensorInfo = [&dataType](const InputSlot& slot)
1092 {
1093 return OverrideDataType(slot.GetConnectedOutputSlot()->GetTensorInfo(), dataType);
1094 };
Finn Williams3e54d032020-10-22 16:53:35 +01001095 auto beginI = MakeTransformIterator(layer.GetInputSlots().begin(), getTensorInfo);
1096 auto endI = MakeTransformIterator(layer.GetInputSlots().end(), getTensorInfo);
Matthew Jackson2b8c1da2019-07-04 14:59:16 +01001097 std::vector<TensorInfo> inputs(beginI, endI);
1098
1099 auto getTensorInfoPtr = [](const TensorInfo& info)
1100 {
1101 return &info;
1102 };
Finn Williams3e54d032020-10-22 16:53:35 +01001103 auto beginPtr = MakeTransformIterator(inputs.begin(), getTensorInfoPtr);
1104 auto endPtr = MakeTransformIterator(inputs.end(), getTensorInfoPtr);
Matthew Jackson2b8c1da2019-07-04 14:59:16 +01001105 std::vector<const TensorInfo*> inputPtrs(beginPtr, endPtr);
1106
1107 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
1108
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001109 result = layerSupportObject.IsStackSupported(inputPtrs, output, cLayer->GetParameters(), reason);
Matthew Jackson2b8c1da2019-07-04 14:59:16 +01001110
1111 break;
1112 }
Derek Lamberti013c3902019-10-21 10:46:16 +01001113 case LayerType::StandIn:
1114 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001115 auto cLayer = PolymorphicDowncast<const StandInLayer*>(&layer);
Derek Lamberti013c3902019-10-21 10:46:16 +01001116
1117 // Get vector of all inputs.
1118 auto getTensorInfoIn = [&dataType](const InputSlot& slot)
1119 {
1120 return OverrideDataType(slot.GetConnectedOutputSlot()->GetTensorInfo(), dataType);
1121 };
1122 auto getTensorInfoOut = [&dataType](const OutputSlot& slot)
1123 {
1124 return OverrideDataType(slot.GetTensorInfo(), dataType);
1125 };
Finn Williams3e54d032020-10-22 16:53:35 +01001126 auto beginI = MakeTransformIterator(layer.GetInputSlots().begin(), getTensorInfoIn);
1127 auto endI = MakeTransformIterator(layer.GetInputSlots().end(), getTensorInfoIn);
Derek Lamberti013c3902019-10-21 10:46:16 +01001128 std::vector<TensorInfo> inputs(beginI, endI);
1129
Finn Williams3e54d032020-10-22 16:53:35 +01001130 auto beginO = MakeTransformIterator(layer.GetOutputSlots().begin(), getTensorInfoOut);
1131 auto endO = MakeTransformIterator(layer.GetOutputSlots().end(), getTensorInfoOut);
Derek Lamberti013c3902019-10-21 10:46:16 +01001132 std::vector<TensorInfo> outputs(beginO, endO);
1133
1134
1135 auto getTensorInfoPtr = [](const TensorInfo& info)
1136 {
1137 return &info;
1138 };
Finn Williams3e54d032020-10-22 16:53:35 +01001139 auto beginPtrI = MakeTransformIterator(inputs.begin(), getTensorInfoPtr);
1140 auto endPtrI = MakeTransformIterator(inputs.end(), getTensorInfoPtr);
Derek Lamberti013c3902019-10-21 10:46:16 +01001141 std::vector<const TensorInfo*> inputPtrs(beginPtrI, endPtrI);
1142
Finn Williams3e54d032020-10-22 16:53:35 +01001143 auto beginPtrO = MakeTransformIterator(outputs.begin(), getTensorInfoPtr);
1144 auto endPtrO = MakeTransformIterator(outputs.end(), getTensorInfoPtr);
Derek Lamberti013c3902019-10-21 10:46:16 +01001145 std::vector<const TensorInfo*> outputPtrs(beginPtrO, endPtrO);
1146
1147
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001148 result = layerSupportObject.IsStandInSupported(inputPtrs,
1149 outputPtrs,
1150 cLayer->GetParameters(),
1151 reason);
Derek Lamberti013c3902019-10-21 10:46:16 +01001152 break;
1153 }
Conor Kennedy430b5d82018-11-14 15:28:28 +00001154 case LayerType::StridedSlice:
1155 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001156 auto cLayer = PolymorphicDowncast<const StridedSliceLayer*>(&layer);
Conor Kennedy430b5d82018-11-14 15:28:28 +00001157 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1158 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001159 result = layerSupportObject.IsStridedSliceSupported(OverrideDataType(input, dataType),
1160 OverrideDataType(output, dataType),
1161 cLayer->GetParameters(),
1162 reason);
Conor Kennedy430b5d82018-11-14 15:28:28 +00001163 break;
1164 }
David Beckc2044fe2018-09-05 15:00:38 +01001165 case LayerType::Subtraction:
1166 {
1167 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1168 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
1169 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001170 result = layerSupportObject.IsSubtractionSupported(
David Beckc2044fe2018-09-05 15:00:38 +01001171 OverrideDataType(input0, dataType),
1172 OverrideDataType(input1, dataType),
1173 OverrideDataType(output, dataType),
David Beck33f0ae02018-10-18 15:13:56 +01001174 reason);
David Beckc2044fe2018-09-05 15:00:38 +01001175 break;
1176 }
Sadik Armaganeff363d2019-04-05 15:25:46 +01001177 case LayerType::Switch:
1178 {
1179 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1180 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
1181 const TensorInfo& output0 = layer.GetOutputSlot(0).GetTensorInfo();
1182 const TensorInfo& output1 = layer.GetOutputSlot(1).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001183 result = layerSupportObject.IsSwitchSupported(OverrideDataType(input0, dataType),
1184 OverrideDataType(input1, dataType),
1185 OverrideDataType(output0, dataType),
1186 OverrideDataType(output1, dataType),
1187 reason);
Sadik Armaganeff363d2019-04-05 15:25:46 +01001188 break;
1189 }
narpra0132b90462018-09-13 11:07:48 +01001190 case LayerType::Mean:
1191 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001192 auto cLayer = PolymorphicDowncast<const MeanLayer*>(&layer);
narpra0132b90462018-09-13 11:07:48 +01001193 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1194 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001195 result = layerSupportObject.IsMeanSupported(
narpra0132b90462018-09-13 11:07:48 +01001196 OverrideDataType(input, dataType),
1197 OverrideDataType(output, dataType),
1198 cLayer->GetParameters(),
David Beck33f0ae02018-10-18 15:13:56 +01001199 reason);
narpra0132b90462018-09-13 11:07:48 +01001200 break;
1201 }
kevmay0190539692018-11-29 08:40:19 +00001202 case LayerType::Minimum:
1203 {
1204 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1205 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
1206 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001207 result = layerSupportObject.IsMinimumSupported(OverrideDataType(input0, dataType),
1208 OverrideDataType(input1, dataType),
1209 OverrideDataType(output, dataType),
1210 reason);
kevmay0190539692018-11-29 08:40:19 +00001211 break;
1212 }
Matteo Martincigh0e406ee2019-06-12 15:42:18 +01001213 case LayerType::Prelu:
1214 {
1215 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1216 const TensorInfo& alpha = layer.GetInputSlot(1).GetConnection()->GetTensorInfo();
1217 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001218 result = layerSupportObject.IsPreluSupported(OverrideDataType(input, dataType),
1219 OverrideDataType(alpha, dataType),
1220 OverrideDataType(output, dataType),
1221 reason);
Matteo Martincigh0e406ee2019-06-12 15:42:18 +01001222 break;
1223 }
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001224 case LayerType::Transpose:
1225 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001226 auto cLayer = PolymorphicDowncast<const TransposeLayer*>(&layer);
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001227 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1228 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001229 result = layerSupportObject.IsTransposeSupported(OverrideDataType(input, dataType),
1230 OverrideDataType(output, dataType),
1231 cLayer->GetParameters(),
1232 reason);
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001233 break;
1234 }
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001235 case LayerType::TransposeConvolution2d:
1236 {
Jan Eilersbb446e52020-04-02 13:56:54 +01001237 auto cLayer = PolymorphicDowncast<const TransposeConvolution2dLayer*>(&layer);
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001238
1239 const TensorInfo input = OverrideDataType(layer.GetInputSlot(0).GetConnection()->GetTensorInfo(),
1240 dataType);
1241 const TensorInfo output = OverrideDataType(layer.GetOutputSlot(0).GetTensorInfo(), dataType);
1242
1243 const TransposeConvolution2dDescriptor& descriptor = cLayer->GetParameters();
1244
1245 Optional<TensorInfo> biases;
1246 if (descriptor.m_BiasEnabled)
1247 {
Narumol Prangnawaratac2770a2020-04-01 16:51:23 +01001248 ARMNN_ASSERT(cLayer->m_Bias.get() != nullptr);
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001249 biases = OverrideDataType(cLayer->m_Bias->GetTensorInfo(),
1250 GetBiasTypeFromWeightsType(dataType));
1251 }
1252
Narumol Prangnawaratac2770a2020-04-01 16:51:23 +01001253 ARMNN_ASSERT(cLayer->m_Weight.get() != nullptr);
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001254 const TensorInfo weights = OverrideDataType(cLayer->m_Weight->GetTensorInfo(), dataType);
1255
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001256 result = layerSupportObject.IsTransposeConvolution2dSupported(input,
1257 output,
1258 descriptor,
1259 weights,
1260 biases,
1261 reason);
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001262
1263 break;
1264 }
Sadik Armagan0c3ea5b2021-02-03 09:29:30 +00001265 case LayerType::Reduce:
1266 {
1267 auto cLayer = PolymorphicDowncast<const ReduceLayer*>(&layer);
1268 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
1269 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
1270
Sadik Armaganf0a6dec2021-03-25 07:46:55 +00001271 result = layerSupportObject.IsReduceSupported(OverrideDataType(input, dataType),
1272 OverrideDataType(output, dataType),
1273 cLayer->GetParameters(),
1274 reason);
Sadik Armagan0c3ea5b2021-02-03 09:29:30 +00001275 break;
1276 }
telsoa014fcda012018-03-09 14:13:49 +00001277 default:
1278 {
Narumol Prangnawaratac2770a2020-04-01 16:51:23 +01001279 ARMNN_ASSERT_MSG(false, "WorkloadFactory did not recognise type of layer.");
David Beck33f0ae02018-10-18 15:13:56 +01001280 reason.value() = "Unrecognised layer type";
telsoa014fcda012018-03-09 14:13:49 +00001281 result = false;
1282 break;
1283 }
1284 }
telsoa014fcda012018-03-09 14:13:49 +00001285 return result;
1286}
1287
Sadik Armagan045f6be2020-09-10 13:37:32 +01001288bool IWorkloadFactory::IsLayerSupported(const BackendId& backendId,
1289 const IConnectableLayer& connectableLayer,
1290 Optional<DataType> dataType,
1291 std::string& outReasonIfUnsupported)
1292{
1293 return IsLayerConfigurationSupported(backendId, connectableLayer, dataType, outReasonIfUnsupported);
1294}
1295
David Beckdcb751f2018-10-03 11:42:42 +01001296bool IWorkloadFactory::IsLayerSupported(const IConnectableLayer& connectableLayer,
David Beck29c75de2018-10-23 13:35:58 +01001297 Optional<DataType> dataType,
telsoa01c577f2c2018-08-31 09:22:23 +01001298 std::string& outReasonIfUnsupported)
telsoa014fcda012018-03-09 14:13:49 +00001299{
Jan Eilersbb446e52020-04-02 13:56:54 +01001300 auto layer = PolymorphicDowncast<const Layer*>(&connectableLayer);
Sadik Armagan045f6be2020-09-10 13:37:32 +01001301 return IsLayerConfigurationSupported(layer->GetBackendId(), connectableLayer, dataType, outReasonIfUnsupported);
1302}
1303
1304// TODO merge with defaulted modelOptions above
1305bool IWorkloadFactory::IsLayerSupported(const IConnectableLayer& connectableLayer,
1306 Optional<DataType> dataType,
1307 std::string& outReasonIfUnsupported,
1308 const ModelOptions& modelOptions)
1309{
1310 auto layer = PolymorphicDowncast<const Layer*>(&connectableLayer);
1311 return IsLayerConfigurationSupported(layer->GetBackendId(),
1312 connectableLayer,
1313 dataType,
1314 outReasonIfUnsupported,
1315 modelOptions);
telsoa014fcda012018-03-09 14:13:49 +00001316}
1317
Sadik Armagan04a72972020-09-14 15:44:18 +01001318bool IWorkloadFactory::IsLayerSupported(const BackendId& backendId,
1319 const IConnectableLayer& connectableLayer,
1320 Optional<DataType> dataType,
1321 std::string& outReasonIfUnsupported,
1322 const ModelOptions& modelOptions)
1323{
1324 return IsLayerConfigurationSupported(backendId,
1325 connectableLayer,
1326 dataType,
1327 outReasonIfUnsupported,
1328 modelOptions);
1329}
1330
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001331// Default Implementations
Derek Lamberti901ea112019-12-10 22:07:09 +00001332std::unique_ptr<IWorkload> IWorkloadFactory::CreateAbs(const AbsQueueDescriptor& /*descriptor*/,
1333 const WorkloadInfo& /*info*/) const
Kevin May868eb142019-09-04 17:29:31 +01001334{
1335 return std::unique_ptr<IWorkload>();
1336}
1337
Derek Lamberti901ea112019-12-10 22:07:09 +00001338std::unique_ptr<IWorkload> IWorkloadFactory::CreateActivation(const ActivationQueueDescriptor& /*descriptor*/,
1339 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001340{
1341 return std::unique_ptr<IWorkload>();
1342}
1343
Derek Lamberti901ea112019-12-10 22:07:09 +00001344std::unique_ptr<IWorkload> IWorkloadFactory::CreateAddition(const AdditionQueueDescriptor& /*descriptor*/,
1345 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001346{
1347 return std::unique_ptr<IWorkload>();
1348}
1349
Derek Lamberti901ea112019-12-10 22:07:09 +00001350std::unique_ptr<IWorkload> IWorkloadFactory::CreateArgMinMax(const ArgMinMaxQueueDescriptor& /*descriptor*/,
1351 const WorkloadInfo& /*info*/) const
Nikhil Rajee391d52019-09-05 17:50:44 +01001352{
1353 return std::unique_ptr<IWorkload>();
1354}
1355
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001356std::unique_ptr<IWorkload> IWorkloadFactory::CreateBatchNormalization(
Derek Lamberti901ea112019-12-10 22:07:09 +00001357 const BatchNormalizationQueueDescriptor& /*descriptor*/, const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001358{
1359 return std::unique_ptr<IWorkload>();
1360}
1361
Derek Lamberti901ea112019-12-10 22:07:09 +00001362std::unique_ptr<IWorkload> IWorkloadFactory::CreateBatchToSpaceNd(const BatchToSpaceNdQueueDescriptor& /*desc*/,
1363 const WorkloadInfo& /*Info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001364{
1365 return std::unique_ptr<IWorkload>();
1366}
1367
mathad01b392e982021-04-07 12:07:30 +01001368std::unique_ptr<IWorkload> IWorkloadFactory::CreateCast(const CastQueueDescriptor& /*descriptor*/,
1369 const WorkloadInfo& /*info*/) const
1370{
1371 return std::unique_ptr<IWorkload>();
1372}
1373
Derek Lamberti901ea112019-12-10 22:07:09 +00001374std::unique_ptr<IWorkload> IWorkloadFactory::CreateComparison(const ComparisonQueueDescriptor& /*descriptor*/,
1375 const WorkloadInfo& /*info*/) const
Aron Virginas-Tar77bfb5e2019-10-16 17:45:38 +01001376{
1377 return std::unique_ptr<IWorkload>();
1378}
1379
Derek Lamberti901ea112019-12-10 22:07:09 +00001380std::unique_ptr<IWorkload> IWorkloadFactory::CreateConcat(const ConcatQueueDescriptor& /*descriptor*/,
1381 const WorkloadInfo& /*info*/) const
Jim Flynn4ed6c832019-05-20 11:02:46 +01001382{
1383 return std::unique_ptr<IWorkload>();
1384}
1385
Derek Lamberti901ea112019-12-10 22:07:09 +00001386std::unique_ptr<IWorkload> IWorkloadFactory::CreateConstant(const ConstantQueueDescriptor& /*descriptor*/,
1387 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001388{
1389 return std::unique_ptr<IWorkload>();
1390}
1391
Narumol Prangnawarat7ddbbae2020-03-13 10:26:05 +00001392std::unique_ptr<IWorkload> IWorkloadFactory::CreateConvertBf16ToFp32(const ConvertBf16ToFp32QueueDescriptor& /*desc*/,
1393 const WorkloadInfo& /*info*/) const
1394{
1395 return std::unique_ptr<IWorkload>();
1396}
1397
Derek Lamberti901ea112019-12-10 22:07:09 +00001398std::unique_ptr<IWorkload> IWorkloadFactory::CreateConvertFp16ToFp32(const ConvertFp16ToFp32QueueDescriptor& /*desc*/,
1399 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001400{
1401 return std::unique_ptr<IWorkload>();
1402}
1403
Narumol Prangnawaratea54a012020-03-16 16:36:10 +00001404std::unique_ptr<IWorkload> IWorkloadFactory::CreateConvertFp32ToBf16(const ConvertFp32ToBf16QueueDescriptor& /*desc*/,
1405 const WorkloadInfo& /*info*/) const
1406{
1407 return std::unique_ptr<IWorkload>();
1408}
1409
Derek Lamberti901ea112019-12-10 22:07:09 +00001410std::unique_ptr<IWorkload> IWorkloadFactory::CreateConvertFp32ToFp16(const ConvertFp32ToFp16QueueDescriptor& /*desc*/,
1411 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001412{
1413 return std::unique_ptr<IWorkload>();
1414}
1415
Derek Lamberti901ea112019-12-10 22:07:09 +00001416std::unique_ptr<IWorkload> IWorkloadFactory::CreateConvolution2d(const Convolution2dQueueDescriptor& /*descriptor*/,
1417 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001418{
1419 return std::unique_ptr<IWorkload>();
1420}
1421
Derek Lamberti901ea112019-12-10 22:07:09 +00001422std::unique_ptr<IWorkload> IWorkloadFactory::CreateDebug(const DebugQueueDescriptor& /*descriptor*/,
1423 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001424{
1425 return std::unique_ptr<IWorkload>();
1426}
1427
Derek Lamberti901ea112019-12-10 22:07:09 +00001428std::unique_ptr<IWorkload> IWorkloadFactory::CreateDepthToSpace(const DepthToSpaceQueueDescriptor& /*descriptor*/,
1429 const WorkloadInfo& /*info*/) const
Aron Virginas-Tardd6247f2019-09-19 14:31:17 +01001430{
1431 return std::unique_ptr<IWorkload>();
1432}
1433
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001434std::unique_ptr<IWorkload> IWorkloadFactory::CreateDepthwiseConvolution2d(
Derek Lamberti901ea112019-12-10 22:07:09 +00001435 const DepthwiseConvolution2dQueueDescriptor& /*descriptor*/, const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001436{
1437 return std::unique_ptr<IWorkload>();
1438}
1439
Nattapat Chaimanowonge4294fd2019-03-28 09:56:53 +00001440std::unique_ptr<IWorkload> IWorkloadFactory::CreateDequantize(
Derek Lamberti901ea112019-12-10 22:07:09 +00001441 const DequantizeQueueDescriptor& /*descriptor*/, const WorkloadInfo& /*info*/) const
Nattapat Chaimanowonge4294fd2019-03-28 09:56:53 +00001442{
1443 return std::unique_ptr<IWorkload>();
1444}
1445
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001446std::unique_ptr<IWorkload> IWorkloadFactory::CreateDetectionPostProcess(
Derek Lamberti901ea112019-12-10 22:07:09 +00001447 const DetectionPostProcessQueueDescriptor& /*descriptor*/, const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001448{
1449 return std::unique_ptr<IWorkload>();
1450}
1451
Derek Lamberti901ea112019-12-10 22:07:09 +00001452std::unique_ptr<IWorkload> IWorkloadFactory::CreateDivision(const DivisionQueueDescriptor& /*descriptor*/,
1453 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001454{
1455 return std::unique_ptr<IWorkload>();
1456}
1457
josh minor4a3c6102020-01-06 16:40:46 -06001458std::unique_ptr<IWorkload> IWorkloadFactory::CreateElementwiseUnary(const ElementwiseUnaryQueueDescriptor& /*desc*/,
1459 const WorkloadInfo& /*info*/) const
1460{
1461 return std::unique_ptr<IWorkload>();
1462}
1463
Derek Lamberti901ea112019-12-10 22:07:09 +00001464std::unique_ptr<IWorkload> IWorkloadFactory::CreateEqual(const EqualQueueDescriptor& /*descriptor*/,
1465 const WorkloadInfo& /*Info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001466{
1467 return std::unique_ptr<IWorkload>();
1468}
1469
Derek Lamberti901ea112019-12-10 22:07:09 +00001470std::unique_ptr<IWorkload> IWorkloadFactory::CreateFakeQuantization(const FakeQuantizationQueueDescriptor& /*desc*/,
1471 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001472{
1473 return std::unique_ptr<IWorkload>();
1474}
1475
Ryan OSheaec6c6802020-06-05 17:17:06 +01001476std::unique_ptr<IWorkload> IWorkloadFactory::CreateFill(const FillQueueDescriptor& /*descriptor*/,
1477 const WorkloadInfo& /*info*/) const
1478{
1479 return std::unique_ptr<IWorkload>();
1480}
1481
Derek Lamberti901ea112019-12-10 22:07:09 +00001482std::unique_ptr<IWorkload> IWorkloadFactory::CreateFloor(const FloorQueueDescriptor& /*descriptor*/,
1483 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001484{
1485 return std::unique_ptr<IWorkload>();
1486}
1487
Derek Lamberti901ea112019-12-10 22:07:09 +00001488std::unique_ptr<IWorkload> IWorkloadFactory::CreateFullyConnected(const FullyConnectedQueueDescriptor& /*descriptor*/,
1489 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001490{
1491 return std::unique_ptr<IWorkload>();
1492}
1493
Derek Lamberti901ea112019-12-10 22:07:09 +00001494std::unique_ptr<IWorkload> IWorkloadFactory::CreateGather(const GatherQueueDescriptor& /*descriptor*/,
1495 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001496{
1497 return std::unique_ptr<IWorkload>();
1498}
1499
Derek Lamberti901ea112019-12-10 22:07:09 +00001500std::unique_ptr<IWorkload> IWorkloadFactory::CreateGreater(const GreaterQueueDescriptor& /*descriptor*/,
1501 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001502{
1503 return std::unique_ptr<IWorkload>();
1504}
1505
Kevin Mayce5045a2019-10-02 14:07:47 +01001506std::unique_ptr<IWorkload> IWorkloadFactory::CreateInstanceNormalization(
Derek Lamberti901ea112019-12-10 22:07:09 +00001507 const InstanceNormalizationQueueDescriptor& /*descriptor*/,
1508 const WorkloadInfo& /*info*/) const
Kevin Mayce5045a2019-10-02 14:07:47 +01001509{
1510 return std::unique_ptr<IWorkload>();
1511}
1512
Derek Lamberti901ea112019-12-10 22:07:09 +00001513std::unique_ptr<IWorkload> IWorkloadFactory::CreateL2Normalization(const L2NormalizationQueueDescriptor& /*desc*/,
1514 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001515{
1516 return std::unique_ptr<IWorkload>();
1517}
1518
James Conroyaba90cd2020-11-06 16:28:18 +00001519std::unique_ptr<IWorkload> IWorkloadFactory::CreateLogicalBinary(const LogicalBinaryQueueDescriptor& /*desc*/,
1520 const WorkloadInfo& /*info*/) const
1521{
1522 return std::unique_ptr<IWorkload>();
1523}
1524
1525std::unique_ptr<IWorkload> IWorkloadFactory::CreateLogicalUnary(const ElementwiseUnaryQueueDescriptor& /*desc*/,
1526 const WorkloadInfo& /*info*/) const
1527{
1528 return std::unique_ptr<IWorkload>();
1529}
1530
Derek Lamberti901ea112019-12-10 22:07:09 +00001531std::unique_ptr<IWorkload> IWorkloadFactory::CreateLogSoftmax(const LogSoftmaxQueueDescriptor& /*descriptor*/,
1532 const WorkloadInfo& /*info*/) const
Aron Virginas-Tarf982dea2019-10-11 14:07:53 +01001533{
1534 return std::unique_ptr<IWorkload>();
1535}
1536
Derek Lamberti901ea112019-12-10 22:07:09 +00001537std::unique_ptr<IWorkload> IWorkloadFactory::CreateLstm(const LstmQueueDescriptor& /*descriptor*/,
1538 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001539{
1540 return std::unique_ptr<IWorkload>();
1541}
1542
Derek Lamberti901ea112019-12-10 22:07:09 +00001543std::unique_ptr<IWorkload> IWorkloadFactory::CreateMaximum(const MaximumQueueDescriptor& /*descriptor*/,
1544 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001545{
1546 return std::unique_ptr<IWorkload>();
1547}
1548
Derek Lamberti901ea112019-12-10 22:07:09 +00001549std::unique_ptr<IWorkload> IWorkloadFactory::CreateMean(const MeanQueueDescriptor& /*descriptor*/,
1550 const WorkloadInfo& /*Info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001551{
1552 return std::unique_ptr<IWorkload>();
1553}
1554
Derek Lamberti901ea112019-12-10 22:07:09 +00001555std::unique_ptr<IWorkload> IWorkloadFactory::CreateMemCopy(const MemCopyQueueDescriptor& /*descriptor*/,
1556 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001557{
1558 return std::unique_ptr<IWorkload>();
1559}
1560
Derek Lamberti901ea112019-12-10 22:07:09 +00001561std::unique_ptr<IWorkload> IWorkloadFactory::CreateMemImport(const MemImportQueueDescriptor& /*descriptor*/,
1562 const WorkloadInfo& /*info*/) const
Derek Lambertif674aa02019-08-01 15:56:25 +01001563{
1564 return std::unique_ptr<IWorkload>();
1565}
1566
Derek Lamberti901ea112019-12-10 22:07:09 +00001567std::unique_ptr<IWorkload> IWorkloadFactory::CreateMerge(const MergeQueueDescriptor& /*descriptor*/,
1568 const WorkloadInfo& /*info*/) const
Nattapat Chaimanowong1f886302019-04-05 13:37:19 +01001569{
1570 return std::unique_ptr<IWorkload>();
1571}
1572
Derek Lamberti901ea112019-12-10 22:07:09 +00001573std::unique_ptr<IWorkload> IWorkloadFactory::CreateMerger(const MergerQueueDescriptor& /*descriptor*/,
1574 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001575{
1576 return std::unique_ptr<IWorkload>();
1577}
1578
Derek Lamberti901ea112019-12-10 22:07:09 +00001579std::unique_ptr<IWorkload> IWorkloadFactory::CreateMinimum(const MinimumQueueDescriptor& /*descriptor*/,
1580 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001581{
1582 return std::unique_ptr<IWorkload>();
1583}
1584
Derek Lamberti901ea112019-12-10 22:07:09 +00001585std::unique_ptr<IWorkload> IWorkloadFactory::CreateMultiplication(const MultiplicationQueueDescriptor& /*descriptor*/,
1586 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001587{
1588 return std::unique_ptr<IWorkload>();
1589}
1590
Derek Lamberti901ea112019-12-10 22:07:09 +00001591std::unique_ptr<IWorkload> IWorkloadFactory::CreateNormalization(const NormalizationQueueDescriptor& /*descriptor*/,
1592 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001593{
1594 return std::unique_ptr<IWorkload>();
1595}
1596
Derek Lamberti901ea112019-12-10 22:07:09 +00001597std::unique_ptr<IWorkload> IWorkloadFactory::CreateOutput(const OutputQueueDescriptor& /*descriptor*/,
1598 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001599{
1600 return std::unique_ptr<IWorkload>();
1601}
1602
Derek Lamberti901ea112019-12-10 22:07:09 +00001603std::unique_ptr<IWorkload> IWorkloadFactory::CreatePad(const PadQueueDescriptor& /*descriptor*/,
1604 const WorkloadInfo& /*Info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001605{
1606 return std::unique_ptr<IWorkload>();
1607}
1608
Derek Lamberti901ea112019-12-10 22:07:09 +00001609std::unique_ptr<IWorkload> IWorkloadFactory::CreatePermute(const PermuteQueueDescriptor& /*descriptor*/,
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001610 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001611{
1612 return std::unique_ptr<IWorkload>();
1613}
1614
Derek Lamberti901ea112019-12-10 22:07:09 +00001615std::unique_ptr<IWorkload> IWorkloadFactory::CreatePooling2d(const Pooling2dQueueDescriptor& /*descriptor*/,
1616 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001617{
1618 return std::unique_ptr<IWorkload>();
1619}
1620
Derek Lamberti901ea112019-12-10 22:07:09 +00001621std::unique_ptr<IWorkload> IWorkloadFactory::CreatePreCompiled(const PreCompiledQueueDescriptor& /*descriptor*/,
1622 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001623{
1624 return std::unique_ptr<IWorkload>();
1625}
1626
Derek Lamberti901ea112019-12-10 22:07:09 +00001627std::unique_ptr<IWorkload> IWorkloadFactory::CreatePrelu(const PreluQueueDescriptor &/*descriptor*/,
1628 const WorkloadInfo &/*info*/) const
Matteo Martincigh0e406ee2019-06-12 15:42:18 +01001629{
1630 return std::unique_ptr<IWorkload>();
1631}
1632
Derek Lamberti901ea112019-12-10 22:07:09 +00001633std::unique_ptr<IWorkload> IWorkloadFactory::CreateQuantize(const QuantizeQueueDescriptor& /*descriptor*/,
1634 const WorkloadInfo& /*Info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001635{
1636 return std::unique_ptr<IWorkload>();
1637}
1638
James Conroy586a9aa2020-03-20 08:49:33 +00001639std::unique_ptr<IWorkload> IWorkloadFactory::CreateQLstm(const QLstmQueueDescriptor& /*descriptor*/,
1640 const WorkloadInfo& /*info*/) const
1641{
1642 return std::unique_ptr<IWorkload>();
1643}
1644
Derek Lamberti901ea112019-12-10 22:07:09 +00001645std::unique_ptr<IWorkload> IWorkloadFactory::CreateQuantizedLstm(const QuantizedLstmQueueDescriptor& /*descriptor*/,
1646 const WorkloadInfo& /*info*/) const
James Conroyee18dc82019-07-17 11:27:46 +01001647{
1648 return std::unique_ptr<IWorkload>();
1649}
Finn Williams2605b232020-06-10 15:53:46 +01001650std::unique_ptr<IWorkload> IWorkloadFactory::CreateRank(const RankQueueDescriptor& /*descriptor*/,
1651 const WorkloadInfo& /*info*/) const
1652{
1653 return std::unique_ptr<IWorkload>();
1654}
James Conroyee18dc82019-07-17 11:27:46 +01001655
Sadik Armagan0c3ea5b2021-02-03 09:29:30 +00001656std::unique_ptr<IWorkload> IWorkloadFactory::CreateReduce(const ReduceQueueDescriptor& /*descriptor*/,
1657 const WorkloadInfo& /*info*/) const
1658{
1659 return std::unique_ptr<IWorkload>();
1660}
1661
Derek Lamberti901ea112019-12-10 22:07:09 +00001662std::unique_ptr<IWorkload> IWorkloadFactory::CreateReshape(const ReshapeQueueDescriptor& /*descriptor*/,
1663 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001664{
1665 return std::unique_ptr<IWorkload>();
1666}
1667
Derek Lamberti901ea112019-12-10 22:07:09 +00001668std::unique_ptr<IWorkload> IWorkloadFactory::CreateResizeBilinear(const ResizeBilinearQueueDescriptor& /*descriptor*/,
1669 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001670{
1671 return std::unique_ptr<IWorkload>();
1672}
1673
Derek Lamberti901ea112019-12-10 22:07:09 +00001674std::unique_ptr<IWorkload> IWorkloadFactory::CreateResize(const ResizeQueueDescriptor& /*descriptor*/,
1675 const WorkloadInfo& /*info*/) const
Teresa Charlina9075df2019-06-27 15:41:57 +01001676{
1677 return std::unique_ptr<IWorkload>();
1678}
1679
Derek Lamberti901ea112019-12-10 22:07:09 +00001680std::unique_ptr<IWorkload> IWorkloadFactory::CreateRsqrt(const RsqrtQueueDescriptor& /*descriptor*/,
1681 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001682{
1683 return std::unique_ptr<IWorkload>();
1684}
1685
Keith Davis3ae3f972021-05-21 16:33:48 +01001686std::unique_ptr<IWorkload> IWorkloadFactory::CreateShape(const ShapeQueueDescriptor& /*descriptor*/,
1687 const WorkloadInfo& /*info*/) const
1688{
1689 return std::unique_ptr<IWorkload>();
1690}
1691
Derek Lamberti901ea112019-12-10 22:07:09 +00001692std::unique_ptr<IWorkload> IWorkloadFactory::CreateSlice(const SliceQueueDescriptor& /*descriptor*/,
1693 const WorkloadInfo& /*info*/) const
1694{
1695 return std::unique_ptr<IWorkload>();
1696}
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001697
Derek Lamberti901ea112019-12-10 22:07:09 +00001698std::unique_ptr<IWorkload> IWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& /*descriptor*/,
1699 const WorkloadInfo& /*info*/) const
Aron Virginas-Tar636ab402019-09-16 14:27:45 +01001700{
1701 return std::unique_ptr<IWorkload>();
1702}
1703
Derek Lamberti901ea112019-12-10 22:07:09 +00001704std::unique_ptr<IWorkload> IWorkloadFactory::CreateSplitter(const SplitterQueueDescriptor& /*descriptor*/,
1705 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001706{
1707 return std::unique_ptr<IWorkload>();
1708}
1709
Derek Lamberti901ea112019-12-10 22:07:09 +00001710std::unique_ptr<IWorkload> IWorkloadFactory::CreateSpaceToBatchNd(const SpaceToBatchNdQueueDescriptor& /*descriptor*/,
1711 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001712{
1713 return std::unique_ptr<IWorkload>();
1714}
1715
Derek Lamberti901ea112019-12-10 22:07:09 +00001716std::unique_ptr<IWorkload> IWorkloadFactory::CreateSpaceToDepth(const SpaceToDepthQueueDescriptor& /*descriptor*/,
1717 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001718{
1719 return std::unique_ptr<IWorkload>();
1720}
1721
Derek Lamberti901ea112019-12-10 22:07:09 +00001722std::unique_ptr<IWorkload> IWorkloadFactory::CreateStack(const StackQueueDescriptor& /*descriptor*/,
1723 const WorkloadInfo& /*info*/) const
Aron Virginas-Tar972af152019-06-11 14:14:03 +01001724{
1725 return std::unique_ptr<IWorkload>();
1726}
1727
Derek Lamberti901ea112019-12-10 22:07:09 +00001728std::unique_ptr<IWorkload> IWorkloadFactory::CreateStridedSlice(const StridedSliceQueueDescriptor& /*descriptor*/,
1729 const WorkloadInfo& /*info*/) const
Matthew Jackson2b8c1da2019-07-04 14:59:16 +01001730{
1731 return std::unique_ptr<IWorkload>();
1732}
1733
Derek Lamberti901ea112019-12-10 22:07:09 +00001734std::unique_ptr<IWorkload> IWorkloadFactory::CreateSubtraction(const SubtractionQueueDescriptor& /*descriptor*/,
1735 const WorkloadInfo& /*info*/) const
Derek Lambertia9cca6a2019-03-25 15:41:58 +00001736{
1737 return std::unique_ptr<IWorkload>();
1738}
1739
Derek Lamberti901ea112019-12-10 22:07:09 +00001740std::unique_ptr<IWorkload> IWorkloadFactory::CreateSwitch(const SwitchQueueDescriptor& /*descriptor*/,
1741 const WorkloadInfo& /*info*/) const
Sadik Armaganeff363d2019-04-05 15:25:46 +01001742{
1743 return std::unique_ptr<IWorkload>();
1744}
1745
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001746std::unique_ptr<IWorkload> IWorkloadFactory::CreateTranspose(const TransposeQueueDescriptor& /*descriptor*/,
1747 const WorkloadInfo& /*info*/) const
1748{
1749 return std::unique_ptr<IWorkload>();
1750}
1751
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001752std::unique_ptr<IWorkload> IWorkloadFactory::CreateTransposeConvolution2d(
Derek Lamberti901ea112019-12-10 22:07:09 +00001753 const TransposeConvolution2dQueueDescriptor& /*descriptor*/,
1754 const WorkloadInfo& /*info*/) const
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001755{
1756 return std::unique_ptr<IWorkload>();
surmeh013537c2c2018-05-18 16:31:43 +01001757}
Aron Virginas-Tar639fb042019-06-20 14:28:19 +01001758
Aron Virginas-Tar636ab402019-09-16 14:27:45 +01001759} // namepsace armnn