blob: 71614643c30fce17c0827a832d7f7f955fccdacb [file] [log] [blame]
telsoa014fcda012018-03-09 14:13:49 +00001//
2// Copyright © 2017 Arm Ltd. All rights reserved.
David Beckecb56cd2018-09-05 12:52:57 +01003// SPDX-License-Identifier: MIT
telsoa014fcda012018-03-09 14:13:49 +00004//
5#pragma once
6
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +00007#include <Graph.hpp>
8
9#include <backendsCommon/WorkloadFactory.hpp>
telsoa014fcda012018-03-09 14:13:49 +000010
11#include <boost/core/ignore_unused.hpp>
12
13namespace
14{
15armnn::Graph dummyGraph;
16
telsoa01c577f2c2018-08-31 09:22:23 +010017// Make a dummy TensorInfo object.
telsoa014fcda012018-03-09 14:13:49 +000018template<armnn::DataType DataType>
19armnn::TensorInfo MakeDummyTensorInfo()
20{
21 return armnn::TensorInfo({2,2,2,2}, DataType);
22}
23
24
25// Make a dummy WorkloadInfo using a dummy TensorInfo.
26template<armnn::DataType DataType>
27armnn::WorkloadInfo MakeDummyWorkloadInfo(unsigned int numInputs, unsigned int numOutputs)
28{
29 armnn::WorkloadInfo info;
30 for (unsigned int i=0; i < numInputs; i++)
31 {
32 info.m_InputTensorInfos.push_back(MakeDummyTensorInfo<DataType>());
33 }
34 for (unsigned int o=0; o < numOutputs; o++)
35 {
36 info.m_OutputTensorInfos.push_back(MakeDummyTensorInfo<DataType>());
37 }
38 return info;
39}
40
telsoa01c577f2c2018-08-31 09:22:23 +010041// Template class to create a dummy layer (2 parameters).
telsoa014fcda012018-03-09 14:13:49 +000042template<typename LayerType, typename DescType = typename LayerType::DescriptorType>
43struct DummyLayer
44{
45 DummyLayer()
46 {
47 m_Layer = dummyGraph.AddLayer<LayerType>(DescType(), "");
48 }
49 ~DummyLayer()
50 {
51 dummyGraph.EraseLayer(m_Layer);
52 }
53 LayerType* m_Layer;
54};
55
telsoa01c577f2c2018-08-31 09:22:23 +010056// Template class to create a dummy layer (1 parameter).
telsoa014fcda012018-03-09 14:13:49 +000057template<typename LayerType>
58struct DummyLayer<LayerType, void>
59{
60 DummyLayer()
61 {
62 m_Layer = dummyGraph.AddLayer<LayerType>("");
63 }
64 ~DummyLayer()
65 {
66 dummyGraph.EraseLayer(m_Layer);
67 }
68 LayerType* m_Layer;
69};
70
71template<>
telsoa01c577f2c2018-08-31 09:22:23 +010072struct DummyLayer<armnn::BatchNormalizationLayer>
73{
74 DummyLayer()
75 {
76 m_Layer = dummyGraph.AddLayer<armnn::BatchNormalizationLayer>(armnn::BatchNormalizationDescriptor(), "");
77 m_Layer->m_Mean = std::make_unique<armnn::ScopedCpuTensorHandle>(
78 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
79 m_Layer->m_Variance = std::make_unique<armnn::ScopedCpuTensorHandle>(
80 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
81 m_Layer->m_Beta = std::make_unique<armnn::ScopedCpuTensorHandle>(
82 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
83 m_Layer->m_Gamma = std::make_unique<armnn::ScopedCpuTensorHandle>(
84 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
85 }
86 ~DummyLayer()
87 {
88 dummyGraph.EraseLayer(m_Layer);
89 }
90 armnn::BatchNormalizationLayer* m_Layer;
91
92};
93
94template<>
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +000095struct DummyLayer<armnn::BatchToSpaceNdLayer>
96{
97 DummyLayer()
98 {
99 m_Layer = dummyGraph.AddLayer<armnn::BatchToSpaceNdLayer>(armnn::BatchToSpaceNdDescriptor(), "");
100 }
101 ~DummyLayer()
102 {
103 dummyGraph.EraseLayer(m_Layer);
104 }
105 armnn::BatchToSpaceNdLayer* m_Layer;
106};
107
108template<>
telsoa014fcda012018-03-09 14:13:49 +0000109struct DummyLayer<armnn::ConstantLayer, void>
110{
111 DummyLayer()
112 {
telsoa01c577f2c2018-08-31 09:22:23 +0100113 m_Layer = dummyGraph.AddLayer<armnn::ConstantLayer>("");
telsoa014fcda012018-03-09 14:13:49 +0000114 }
115 ~DummyLayer()
116 {
117 dummyGraph.EraseLayer(m_Layer);
118 }
119 armnn::ConstantLayer* m_Layer;
120};
121
122template<>
123struct DummyLayer<armnn::InputLayer, armnn::LayerBindingId>
124{
125 DummyLayer()
126 {
127 m_Layer = dummyGraph.AddLayer<armnn::InputLayer>(armnn::LayerBindingId(), "");
128
129 }
130 ~DummyLayer()
131 {
132 dummyGraph.EraseLayer(m_Layer);
133 }
134 armnn::InputLayer* m_Layer;
135};
136
137template<>
Jim Flynne242f2d2019-05-22 14:24:13 +0100138struct DummyLayer<armnn::ConcatLayer>
telsoa014fcda012018-03-09 14:13:49 +0000139{
140 DummyLayer()
141 {
142 armnn::OriginsDescriptor desc(2);
Jim Flynne242f2d2019-05-22 14:24:13 +0100143 m_Layer = dummyGraph.AddLayer<armnn::ConcatLayer>(desc, "");
telsoa014fcda012018-03-09 14:13:49 +0000144
145 }
146 ~DummyLayer()
147 {
148 dummyGraph.EraseLayer(m_Layer);
149 }
Jim Flynne242f2d2019-05-22 14:24:13 +0100150 armnn::ConcatLayer* m_Layer;
telsoa014fcda012018-03-09 14:13:49 +0000151};
152
153template<>
154struct DummyLayer<armnn::OutputLayer, armnn::LayerBindingId>
155{
156 DummyLayer()
157 {
158 m_Layer = dummyGraph.AddLayer<armnn::OutputLayer>(armnn::LayerBindingId(), "");
159
160 }
161 ~DummyLayer()
162 {
163 dummyGraph.EraseLayer(m_Layer);
164 }
165 armnn::OutputLayer* m_Layer;
166};
167
168template<>
169struct DummyLayer<armnn::SplitterLayer>
170{
171 DummyLayer()
172 {
173 armnn::ViewsDescriptor desc(1);
174 m_Layer = dummyGraph.AddLayer<armnn::SplitterLayer>(desc, "");
175
176 }
177 ~DummyLayer()
178 {
179 dummyGraph.EraseLayer(m_Layer);
180 }
181 armnn::SplitterLayer* m_Layer;
182};
183
184template <typename ConvolutionLayerType>
185struct DummyConvolutionLayer
186{
187 DummyConvolutionLayer()
188 {
189 typename ConvolutionLayerType::DescriptorType desc;
190 m_Layer = dummyGraph.AddLayer<ConvolutionLayerType>(desc, "");
191 m_Layer->m_Weight = std::make_unique<armnn::ScopedCpuTensorHandle>(
192 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
193 m_Layer->m_Bias = std::make_unique<armnn::ScopedCpuTensorHandle>(
194 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
195 }
196 ~DummyConvolutionLayer()
197 {
198 dummyGraph.EraseLayer(m_Layer);
199 }
200 ConvolutionLayerType* m_Layer;
201};
202
203template<>
204struct DummyLayer<armnn::Convolution2dLayer>
205 : public DummyConvolutionLayer<armnn::Convolution2dLayer>
206{
207};
208
209template<>
210struct DummyLayer<armnn::DepthwiseConvolution2dLayer>
211 : public DummyConvolutionLayer<armnn::DepthwiseConvolution2dLayer>
212{
213};
214
telsoa01c577f2c2018-08-31 09:22:23 +0100215template <typename LstmLayerType>
216struct DummyLstmLayer
217{
218 DummyLstmLayer()
219 {
220 typename LstmLayerType::DescriptorType desc;
221 desc.m_CifgEnabled = false;
222
223 m_Layer = dummyGraph.AddLayer<LstmLayerType>(armnn::LstmDescriptor(), "");
224 m_Layer->m_BasicParameters.m_InputToForgetWeights = std::make_unique<armnn::ScopedCpuTensorHandle>(
225 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
226 m_Layer->m_BasicParameters.m_InputToCellWeights = std::make_unique<armnn::ScopedCpuTensorHandle>(
227 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
228 m_Layer->m_BasicParameters.m_InputToOutputWeights = std::make_unique<armnn::ScopedCpuTensorHandle>(
229 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
230 m_Layer->m_BasicParameters.m_RecurrentToForgetWeights = std::make_unique<armnn::ScopedCpuTensorHandle>(
231 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
232 m_Layer->m_BasicParameters.m_RecurrentToCellWeights = std::make_unique<armnn::ScopedCpuTensorHandle>(
233 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
234 m_Layer->m_BasicParameters.m_RecurrentToOutputWeights = std::make_unique<armnn::ScopedCpuTensorHandle>(
235 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
236 m_Layer->m_BasicParameters.m_ForgetGateBias = std::make_unique<armnn::ScopedCpuTensorHandle>(
237 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
238 m_Layer->m_BasicParameters.m_CellBias = std::make_unique<armnn::ScopedCpuTensorHandle>(
239 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
240 m_Layer->m_BasicParameters.m_OutputGateBias = std::make_unique<armnn::ScopedCpuTensorHandle>(
241 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
242
243 m_Layer->m_CifgParameters.m_InputToInputWeights = std::make_unique<armnn::ScopedCpuTensorHandle>(
244 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
245 m_Layer->m_CifgParameters.m_RecurrentToInputWeights = std::make_unique<armnn::ScopedCpuTensorHandle>(
246 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
247 m_Layer->m_CifgParameters.m_CellToInputWeights = std::make_unique<armnn::ScopedCpuTensorHandle>(
248 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
249 m_Layer->m_CifgParameters.m_InputGateBias = std::make_unique<armnn::ScopedCpuTensorHandle>(
250 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
251 }
252 ~DummyLstmLayer()
253 {
254 dummyGraph.EraseLayer(m_Layer);
255 }
256 armnn::LstmLayer* m_Layer;
257};
258
259template<>
260struct DummyLayer<armnn::LstmLayer>
261 : public DummyLstmLayer<armnn::LstmLayer>
262{
263};
264
265template<>
266struct DummyLayer<armnn::FullyConnectedLayer>
267{
268 DummyLayer()
269 {
270 armnn::FullyConnectedLayer::DescriptorType desc;
271 m_Layer = dummyGraph.AddLayer<armnn::FullyConnectedLayer>(desc, "");
272 m_Layer->m_Weight = std::make_unique<armnn::ScopedCpuTensorHandle>(
273 armnn::TensorInfo(armnn::TensorShape({1,1,1,1}), armnn::DataType::Float32));
274 }
275 ~DummyLayer()
276 {
277 dummyGraph.EraseLayer(m_Layer);
278 }
279 armnn::FullyConnectedLayer* m_Layer;
280};
281
telsoa014fcda012018-03-09 14:13:49 +0000282// Tag for giving LayerType entries a unique strong type each.
283template<armnn::LayerType>
284struct Tag{};
285
286#define DECLARE_LAYER_POLICY_CUSTOM_PARAM(name, descType) \
287template<armnn::DataType DataType> \
288struct LayerTypePolicy<armnn::LayerType::name, DataType> \
289{ \
290 using Type = armnn::name##Layer; \
291 using Desc = descType; \
292 using QueueDesc = armnn::name##QueueDescriptor; \
293 constexpr static const char* NameStr = #name; \
294 \
295 static std::unique_ptr<armnn::IWorkload> MakeDummyWorkload(armnn::IWorkloadFactory *factory, \
296 unsigned int nIn, unsigned int nOut) \
297 { \
298 QueueDesc desc; \
299 armnn::WorkloadInfo info = MakeDummyWorkloadInfo<DataType>(nIn, nOut); \
300 return factory->Create##name(desc, info); \
301 } \
302};
303
telsoa01c577f2c2018-08-31 09:22:23 +0100304// Define a layer policy specialization for use with the IsLayerSupported tests.
telsoa014fcda012018-03-09 14:13:49 +0000305// Use this version for layers whose constructor takes 1 parameter(name).
306#define DECLARE_LAYER_POLICY_1_PARAM(name) DECLARE_LAYER_POLICY_CUSTOM_PARAM(name, void)
307
telsoa01c577f2c2018-08-31 09:22:23 +0100308// Define a layer policy specialization for use with the IsLayerSupported tests.
telsoa014fcda012018-03-09 14:13:49 +0000309// Use this version for layers whose constructor takes 2 parameters(descriptor and name).
310#define DECLARE_LAYER_POLICY_2_PARAM(name) DECLARE_LAYER_POLICY_CUSTOM_PARAM(name, armnn::name##Descriptor)
311
telsoa01c577f2c2018-08-31 09:22:23 +0100312// Layer policy template.
telsoa014fcda012018-03-09 14:13:49 +0000313template<armnn::LayerType Type, armnn::DataType DataType>
314struct LayerTypePolicy;
315
316// Every entry in the armnn::LayerType enum must be accounted for below.
317DECLARE_LAYER_POLICY_2_PARAM(Activation)
318
319DECLARE_LAYER_POLICY_1_PARAM(Addition)
320
321DECLARE_LAYER_POLICY_2_PARAM(BatchNormalization)
322
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000323DECLARE_LAYER_POLICY_2_PARAM(BatchToSpaceNd)
324
Jim Flynne242f2d2019-05-22 14:24:13 +0100325DECLARE_LAYER_POLICY_2_PARAM(Concat)
326
telsoa014fcda012018-03-09 14:13:49 +0000327DECLARE_LAYER_POLICY_1_PARAM(Constant)
328
telsoa01c577f2c2018-08-31 09:22:23 +0100329DECLARE_LAYER_POLICY_1_PARAM(ConvertFp16ToFp32)
330
331DECLARE_LAYER_POLICY_1_PARAM(ConvertFp32ToFp16)
332
telsoa014fcda012018-03-09 14:13:49 +0000333DECLARE_LAYER_POLICY_2_PARAM(Convolution2d)
334
335DECLARE_LAYER_POLICY_1_PARAM(MemCopy)
336
Nattapat Chaimanowong964e9552019-03-26 11:03:26 +0000337DECLARE_LAYER_POLICY_1_PARAM(Debug)
Nattapat Chaimanowonga9a1cf12018-12-03 16:06:49 +0000338
telsoa014fcda012018-03-09 14:13:49 +0000339DECLARE_LAYER_POLICY_2_PARAM(DepthwiseConvolution2d)
340
Nattapat Chaimanowonge4294fd2019-03-28 09:56:53 +0000341DECLARE_LAYER_POLICY_1_PARAM(Dequantize)
342
Narumol Prangnawarat94dd5d82019-01-23 18:06:26 +0000343DECLARE_LAYER_POLICY_2_PARAM(DetectionPostProcess)
344
FrancisMurtagh20995952018-12-17 12:11:36 +0000345DECLARE_LAYER_POLICY_1_PARAM(Equal)
346
telsoa014fcda012018-03-09 14:13:49 +0000347DECLARE_LAYER_POLICY_2_PARAM(FakeQuantization)
348
349DECLARE_LAYER_POLICY_1_PARAM(Floor)
350
351DECLARE_LAYER_POLICY_2_PARAM(FullyConnected)
352
narpra01b89b05f2019-01-16 09:53:09 +0000353DECLARE_LAYER_POLICY_1_PARAM(Gather)
354
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000355DECLARE_LAYER_POLICY_1_PARAM(Greater)
356
telsoa014fcda012018-03-09 14:13:49 +0000357DECLARE_LAYER_POLICY_CUSTOM_PARAM(Input, armnn::LayerBindingId)
358
Matteo Martincighbcd3c852018-09-28 14:14:12 +0100359DECLARE_LAYER_POLICY_2_PARAM(L2Normalization)
telsoa014fcda012018-03-09 14:13:49 +0000360
telsoa01c577f2c2018-08-31 09:22:23 +0100361DECLARE_LAYER_POLICY_2_PARAM(Lstm)
362
Nattapat Chaimanowong5a4304a2018-11-28 10:44:37 +0000363DECLARE_LAYER_POLICY_1_PARAM(Maximum)
364
narpra0132b90462018-09-13 11:07:48 +0100365DECLARE_LAYER_POLICY_2_PARAM(Mean)
366
Nattapat Chaimanowong1f886302019-04-05 13:37:19 +0100367DECLARE_LAYER_POLICY_1_PARAM(Merge)
368
kevmay0190539692018-11-29 08:40:19 +0000369DECLARE_LAYER_POLICY_1_PARAM(Minimum)
370
telsoa014fcda012018-03-09 14:13:49 +0000371DECLARE_LAYER_POLICY_1_PARAM(Multiplication)
372
373DECLARE_LAYER_POLICY_2_PARAM(Normalization)
374
375DECLARE_LAYER_POLICY_CUSTOM_PARAM(Output, armnn::LayerBindingId)
376
Mohamed Nour Abouelseoud5662c202018-09-24 13:30:09 +0100377DECLARE_LAYER_POLICY_2_PARAM(Pad)
378
Derek Lambertia9cca6a2019-03-25 15:41:58 +0000379DECLARE_LAYER_POLICY_1_PARAM(Quantize)
380
telsoa014fcda012018-03-09 14:13:49 +0000381DECLARE_LAYER_POLICY_2_PARAM(Permute)
382
383DECLARE_LAYER_POLICY_2_PARAM(Pooling2d)
384
Matteo Martincigh49124022019-01-11 13:25:59 +0000385DECLARE_LAYER_POLICY_2_PARAM(PreCompiled)
386
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100387DECLARE_LAYER_POLICY_1_PARAM(Division)
388
telsoa014fcda012018-03-09 14:13:49 +0000389DECLARE_LAYER_POLICY_2_PARAM(ResizeBilinear)
390
telsoa01c577f2c2018-08-31 09:22:23 +0100391DECLARE_LAYER_POLICY_2_PARAM(Reshape)
392
Mohamed Nour Abouelseouda1d3c6a2018-12-27 12:39:16 +0000393DECLARE_LAYER_POLICY_1_PARAM(Rsqrt)
394
telsoa014fcda012018-03-09 14:13:49 +0000395DECLARE_LAYER_POLICY_2_PARAM(Softmax)
396
Nattapat Chaimanowong207ef9a2018-11-02 10:57:25 +0000397DECLARE_LAYER_POLICY_2_PARAM(SpaceToBatchNd)
398
telsoa014fcda012018-03-09 14:13:49 +0000399DECLARE_LAYER_POLICY_2_PARAM(Splitter)
400
Conor Kennedy430b5d82018-11-14 15:28:28 +0000401DECLARE_LAYER_POLICY_2_PARAM(StridedSlice)
402
David Beckc2044fe2018-09-05 15:00:38 +0100403DECLARE_LAYER_POLICY_1_PARAM(Subtraction)
telsoa014fcda012018-03-09 14:13:49 +0000404
Sadik Armaganeff363d2019-04-05 15:25:46 +0100405DECLARE_LAYER_POLICY_1_PARAM(Switch)
406
telsoa014fcda012018-03-09 14:13:49 +0000407
408// Generic implementation to get the number of input slots for a given layer type;
409template<armnn::LayerType Type>
410unsigned int GetNumInputs(const armnn::Layer& layer)
411{
412 return layer.GetNumInputSlots();
413}
414
415// Generic implementation to get the number of output slots for a given layer type;
416template<armnn::LayerType Type>
417unsigned int GetNumOutputs(const armnn::Layer& layer)
418{
419 return layer.GetNumOutputSlots();
420}
421
422template<>
Jim Flynne242f2d2019-05-22 14:24:13 +0100423unsigned int GetNumInputs<armnn::LayerType::Concat>(const armnn::Layer& layer)
telsoa014fcda012018-03-09 14:13:49 +0000424{
425 boost::ignore_unused(layer);
426 return 2;
427}
428
telsoa01c577f2c2018-08-31 09:22:23 +0100429// Tests that the IsLayerSupported() function returns the correct value.
430// We determined the correct value by *trying* to create the relevant workload and seeing if it matches what we expect.
telsoa014fcda012018-03-09 14:13:49 +0000431// Returns true if expectations are met, otherwise returns false.
432template<typename FactoryType, armnn::DataType DataType, armnn::LayerType Type>
433bool IsLayerSupportedTest(FactoryType *factory, Tag<Type>)
434{
435 using LayerPolicy = LayerTypePolicy<Type, DataType>;
436 using LayerType = typename LayerPolicy::Type;
437 using LayerDesc = typename LayerPolicy::Desc;
438 DummyLayer<LayerType, LayerDesc> layer;
439
440 unsigned int numIn = GetNumInputs<Type>(*layer.m_Layer);
441 unsigned int numOut = GetNumOutputs<Type>(*layer.m_Layer);
442
telsoa01c577f2c2018-08-31 09:22:23 +0100443 // Make another dummy layer just to make IsLayerSupported have valid inputs.
telsoa014fcda012018-03-09 14:13:49 +0000444 DummyLayer<armnn::ConstantLayer, void> previousLayer;
telsoa01c577f2c2018-08-31 09:22:23 +0100445 // Set output of the previous layer to a dummy tensor.
telsoa014fcda012018-03-09 14:13:49 +0000446 armnn::TensorInfo output = MakeDummyTensorInfo<DataType>();
447 previousLayer.m_Layer->GetOutputSlot(0).SetTensorInfo(output);
telsoa01c577f2c2018-08-31 09:22:23 +0100448 // Connect all outputs of the previous layer to inputs of tested layer.
telsoa014fcda012018-03-09 14:13:49 +0000449 for (unsigned int i = 0; i < numIn; i++)
450 {
451 armnn::IOutputSlot& previousLayerOutputSlot = previousLayer.m_Layer->GetOutputSlot(0);
452 armnn::IInputSlot& layerInputSlot = layer.m_Layer->GetInputSlot(i);
453 previousLayerOutputSlot.Connect(layerInputSlot);
454 }
telsoa01c577f2c2018-08-31 09:22:23 +0100455 // Set outputs of tested layer to a dummy tensor.
telsoa014fcda012018-03-09 14:13:49 +0000456 for (unsigned int i = 0; i < numOut; i++)
457 {
458 layer.m_Layer->GetOutputSlot(0).SetTensorInfo(output);
459 }
460
461 std::string layerName = LayerPolicy::NameStr;
462 std::string reasonIfUnsupported;
463 if (FactoryType::IsLayerSupported(*layer.m_Layer, DataType, reasonIfUnsupported))
464 {
465 std::string errorMsg = " layer expected support but found none.";
466 try
467 {
468 bool retVal = LayerPolicy::MakeDummyWorkload(factory, numIn, numOut).get() != nullptr;
Matteo Martincighfbebcbd2018-10-16 09:45:08 +0100469 BOOST_CHECK_MESSAGE(retVal, layerName << errorMsg);
telsoa014fcda012018-03-09 14:13:49 +0000470 return retVal;
471 }
telsoa01c577f2c2018-08-31 09:22:23 +0100472 catch(const armnn::InvalidArgumentException& e)
telsoa014fcda012018-03-09 14:13:49 +0000473 {
474 boost::ignore_unused(e);
475 // This is ok since we throw InvalidArgumentException when creating the dummy workload.
476 return true;
477 }
478 catch(const std::exception& e)
479 {
480 errorMsg = e.what();
481 BOOST_TEST_ERROR(layerName << ": " << errorMsg);
482 return false;
483 }
telsoa01c577f2c2018-08-31 09:22:23 +0100484 catch(...)
telsoa014fcda012018-03-09 14:13:49 +0000485 {
486 errorMsg = "Unexpected error while testing support for ";
487 BOOST_TEST_ERROR(errorMsg << layerName);
488 return false;
489 }
490 }
491 else
492 {
493 std::string errorMsg = "layer expected no support (giving reason: " + reasonIfUnsupported + ") but found some.";
494 try
495 {
496 bool retVal = LayerPolicy::MakeDummyWorkload(factory, numIn, numOut).get() == nullptr;
497 BOOST_CHECK_MESSAGE(retVal, layerName << errorMsg);
498 return retVal;
499 }
500 // These two exceptions are ok: For workloads that are partially supported, attempting to instantiate them
501 // using parameters that make IsLayerSupported() return false should throw an
telsoa01c577f2c2018-08-31 09:22:23 +0100502 // InvalidArgumentException or UnimplementedException.
telsoa014fcda012018-03-09 14:13:49 +0000503 catch(const armnn::InvalidArgumentException& e)
504 {
505 boost::ignore_unused(e);
506 return true;
507 }
telsoa01c577f2c2018-08-31 09:22:23 +0100508 catch(const armnn::UnimplementedException& e)
telsoa014fcda012018-03-09 14:13:49 +0000509 {
510 boost::ignore_unused(e);
511 return true;
512 }
513 catch(const std::exception& e)
514 {
515 errorMsg = e.what();
516 BOOST_TEST_ERROR(layerName << ": " << errorMsg);
517 return false;
518 }
telsoa01c577f2c2018-08-31 09:22:23 +0100519 catch(...)
telsoa014fcda012018-03-09 14:13:49 +0000520 {
521 errorMsg = "Unexpected error while testing support for ";
522 BOOST_TEST_ERROR(errorMsg << layerName);
523 return false;
524 }
525 }
526}
527
telsoa01c577f2c2018-08-31 09:22:23 +0100528// Helper function to compute the next type in the LayerType enum.
telsoa014fcda012018-03-09 14:13:49 +0000529constexpr armnn::LayerType NextType(armnn::LayerType type)
530{
531 return static_cast<armnn::LayerType>(static_cast<int>(type)+1);
532}
533
telsoa01c577f2c2018-08-31 09:22:23 +0100534// Termination function for determining the end of the LayerType enumeration.
telsoa014fcda012018-03-09 14:13:49 +0000535template<typename FactoryType, armnn::DataType DataType, armnn::LayerType Type>
536bool IsLayerSupportedTestsImpl(FactoryType *factory, Tag<armnn::LayerType::LastLayer>)
537{
538 return IsLayerSupportedTest<FactoryType, DataType, Type>(factory, Tag<Type>());
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000539}
telsoa014fcda012018-03-09 14:13:49 +0000540
telsoa01c577f2c2018-08-31 09:22:23 +0100541// Recursive function to test and enter in the LayerType enum and then iterate on the next entry.
telsoa014fcda012018-03-09 14:13:49 +0000542template<typename FactoryType, armnn::DataType DataType, armnn::LayerType Type>
543bool IsLayerSupportedTestsImpl(FactoryType *factory, Tag<Type>)
544{
545 bool v = IsLayerSupportedTest<FactoryType, DataType, Type>(factory, Tag<Type>());
546
547 return v &&
548 IsLayerSupportedTestsImpl<FactoryType, DataType, NextType(Type)>
549 (factory, Tag<NextType(Type)>());
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000550}
telsoa014fcda012018-03-09 14:13:49 +0000551
552// Helper function to pass through to the test framework.
553template<typename FactoryType, armnn::DataType DataType>
554bool IsLayerSupportedTests(FactoryType *factory)
555{
556 return IsLayerSupportedTestsImpl<FactoryType, DataType>(factory, Tag<armnn::LayerType::FirstLayer>());
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000557}
telsoa014fcda012018-03-09 14:13:49 +0000558
559template<armnn::LayerType Type>
560bool TestLayerTypeMatches()
561{
562 using LayerPolicy = LayerTypePolicy<Type, armnn::DataType::Float32>;
563 using LayerType = typename LayerPolicy::Type;
564 using LayerDesc = typename LayerPolicy::Desc;
565 DummyLayer<LayerType, LayerDesc> layer;
566
567 std::stringstream ss;
568 ss << LayerPolicy::NameStr << " layer type mismatches expected layer type value.";
569 bool v = Type == layer.m_Layer->GetType();
570 BOOST_CHECK_MESSAGE(v, ss.str());
571 return v;
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000572}
telsoa014fcda012018-03-09 14:13:49 +0000573
574template<armnn::LayerType Type>
575bool LayerTypeMatchesTestImpl(Tag<armnn::LayerType::LastLayer>)
576{
577 return TestLayerTypeMatches<Type>();
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000578}
telsoa014fcda012018-03-09 14:13:49 +0000579
580template<armnn::LayerType Type>
581bool LayerTypeMatchesTestImpl(Tag<Type>)
582{
583 return TestLayerTypeMatches<Type>() &&
584 LayerTypeMatchesTestImpl<NextType(Type)>(Tag<NextType(Type)>());
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000585}
telsoa014fcda012018-03-09 14:13:49 +0000586
telsoa01c577f2c2018-08-31 09:22:23 +0100587template<typename FactoryType, typename LayerType, armnn::DataType InputDataType , armnn::DataType OutputDataType>
588bool IsConvertLayerSupportedTests(std::string& reasonIfUnsupported)
589{
590 armnn::Graph graph;
591 LayerType* const layer = graph.AddLayer<LayerType>("LayerName");
592
593 armnn::Layer* const input = graph.AddLayer<armnn::InputLayer>(0, "input");
594 armnn::Layer* const output = graph.AddLayer<armnn::OutputLayer>(0, "output");
595
596 armnn::TensorInfo inputTensorInfo({1, 3, 2, 3}, InputDataType);
597 armnn::TensorInfo outputTensorInfo({1, 3, 2, 3}, OutputDataType);
598
599 input->GetOutputSlot(0).Connect(layer->GetInputSlot(0));
600 input->GetOutputHandler(0).SetTensorInfo(inputTensorInfo);
601 layer->GetOutputSlot(0).Connect(output->GetInputSlot(0));
602 layer->GetOutputHandler(0).SetTensorInfo(outputTensorInfo);
603
604 bool result = FactoryType::IsLayerSupported(*layer, InputDataType, reasonIfUnsupported);
605
606 return result;
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000607}
telsoa01c577f2c2018-08-31 09:22:23 +0100608
Matthew Bentham1f0ff352019-01-02 13:26:31 +0000609template<typename FactoryType, armnn::DataType InputDataType , armnn::DataType OutputDataType>
610bool IsMeanLayerSupportedTests(std::string& reasonIfUnsupported)
611{
612 armnn::Graph graph;
613 static const std::vector<unsigned> axes = {1, 0};
614 armnn::MeanDescriptor desc(axes, false);
615
616 armnn::Layer* const layer = graph.AddLayer<armnn::MeanLayer>(desc, "LayerName");
617
618 armnn::Layer* const input = graph.AddLayer<armnn::InputLayer>(0, "input");
619 armnn::Layer* const output = graph.AddLayer<armnn::OutputLayer>(0, "output");
620
621 armnn::TensorInfo inputTensorInfo({4, 3, 2}, InputDataType);
622 armnn::TensorInfo outputTensorInfo({2}, OutputDataType);
623
624 input->GetOutputSlot(0).Connect(layer->GetInputSlot(0));
625 input->GetOutputHandler(0).SetTensorInfo(inputTensorInfo);
626 layer->GetOutputSlot(0).Connect(output->GetInputSlot(0));
627 layer->GetOutputHandler(0).SetTensorInfo(outputTensorInfo);
628
629 bool result = FactoryType::IsLayerSupported(*layer, InputDataType, reasonIfUnsupported);
630
631 return result;
632}
633
634
telsoa014fcda012018-03-09 14:13:49 +0000635} //namespace