blob: 3df71830caa1febfe3bc76d7f0f694cfd452e8ca [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<>
138struct DummyLayer<armnn::MergerLayer>
139{
140 DummyLayer()
141 {
142 armnn::OriginsDescriptor desc(2);
143 m_Layer = dummyGraph.AddLayer<armnn::MergerLayer>(desc, "");
144
145 }
146 ~DummyLayer()
147 {
148 dummyGraph.EraseLayer(m_Layer);
149 }
150 armnn::MergerLayer* m_Layer;
151};
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
telsoa014fcda012018-03-09 14:13:49 +0000325DECLARE_LAYER_POLICY_1_PARAM(Constant)
326
telsoa01c577f2c2018-08-31 09:22:23 +0100327DECLARE_LAYER_POLICY_1_PARAM(ConvertFp16ToFp32)
328
329DECLARE_LAYER_POLICY_1_PARAM(ConvertFp32ToFp16)
330
telsoa014fcda012018-03-09 14:13:49 +0000331DECLARE_LAYER_POLICY_2_PARAM(Convolution2d)
332
333DECLARE_LAYER_POLICY_1_PARAM(MemCopy)
334
Nattapat Chaimanowongac5aa1f2018-12-05 15:17:18 +0000335DECLARE_LAYER_POLICY_2_PARAM(Debug)
Nattapat Chaimanowonga9a1cf12018-12-03 16:06:49 +0000336
telsoa014fcda012018-03-09 14:13:49 +0000337DECLARE_LAYER_POLICY_2_PARAM(DepthwiseConvolution2d)
338
Narumol Prangnawarat94dd5d82019-01-23 18:06:26 +0000339DECLARE_LAYER_POLICY_2_PARAM(DetectionPostProcess)
340
FrancisMurtagh20995952018-12-17 12:11:36 +0000341DECLARE_LAYER_POLICY_1_PARAM(Equal)
342
telsoa014fcda012018-03-09 14:13:49 +0000343DECLARE_LAYER_POLICY_2_PARAM(FakeQuantization)
344
345DECLARE_LAYER_POLICY_1_PARAM(Floor)
346
347DECLARE_LAYER_POLICY_2_PARAM(FullyConnected)
348
narpra01b89b05f2019-01-16 09:53:09 +0000349DECLARE_LAYER_POLICY_1_PARAM(Gather)
350
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000351DECLARE_LAYER_POLICY_1_PARAM(Greater)
352
telsoa014fcda012018-03-09 14:13:49 +0000353DECLARE_LAYER_POLICY_CUSTOM_PARAM(Input, armnn::LayerBindingId)
354
Matteo Martincighbcd3c852018-09-28 14:14:12 +0100355DECLARE_LAYER_POLICY_2_PARAM(L2Normalization)
telsoa014fcda012018-03-09 14:13:49 +0000356
telsoa01c577f2c2018-08-31 09:22:23 +0100357DECLARE_LAYER_POLICY_2_PARAM(Lstm)
358
Nattapat Chaimanowong5a4304a2018-11-28 10:44:37 +0000359DECLARE_LAYER_POLICY_1_PARAM(Maximum)
360
narpra0132b90462018-09-13 11:07:48 +0100361DECLARE_LAYER_POLICY_2_PARAM(Mean)
362
telsoa014fcda012018-03-09 14:13:49 +0000363DECLARE_LAYER_POLICY_2_PARAM(Merger)
364
kevmay0190539692018-11-29 08:40:19 +0000365DECLARE_LAYER_POLICY_1_PARAM(Minimum)
366
telsoa014fcda012018-03-09 14:13:49 +0000367DECLARE_LAYER_POLICY_1_PARAM(Multiplication)
368
369DECLARE_LAYER_POLICY_2_PARAM(Normalization)
370
371DECLARE_LAYER_POLICY_CUSTOM_PARAM(Output, armnn::LayerBindingId)
372
Mohamed Nour Abouelseoud5662c202018-09-24 13:30:09 +0100373DECLARE_LAYER_POLICY_2_PARAM(Pad)
374
telsoa014fcda012018-03-09 14:13:49 +0000375DECLARE_LAYER_POLICY_2_PARAM(Permute)
376
377DECLARE_LAYER_POLICY_2_PARAM(Pooling2d)
378
Matteo Martincigh49124022019-01-11 13:25:59 +0000379DECLARE_LAYER_POLICY_2_PARAM(PreCompiled)
380
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100381DECLARE_LAYER_POLICY_1_PARAM(Division)
382
telsoa014fcda012018-03-09 14:13:49 +0000383DECLARE_LAYER_POLICY_2_PARAM(ResizeBilinear)
384
telsoa01c577f2c2018-08-31 09:22:23 +0100385DECLARE_LAYER_POLICY_2_PARAM(Reshape)
386
Mohamed Nour Abouelseouda1d3c6a2018-12-27 12:39:16 +0000387DECLARE_LAYER_POLICY_1_PARAM(Rsqrt)
388
telsoa014fcda012018-03-09 14:13:49 +0000389DECLARE_LAYER_POLICY_2_PARAM(Softmax)
390
Nattapat Chaimanowong207ef9a2018-11-02 10:57:25 +0000391DECLARE_LAYER_POLICY_2_PARAM(SpaceToBatchNd)
392
telsoa014fcda012018-03-09 14:13:49 +0000393DECLARE_LAYER_POLICY_2_PARAM(Splitter)
394
Conor Kennedy430b5d82018-11-14 15:28:28 +0000395DECLARE_LAYER_POLICY_2_PARAM(StridedSlice)
396
David Beckc2044fe2018-09-05 15:00:38 +0100397DECLARE_LAYER_POLICY_1_PARAM(Subtraction)
telsoa014fcda012018-03-09 14:13:49 +0000398
399
400// Generic implementation to get the number of input slots for a given layer type;
401template<armnn::LayerType Type>
402unsigned int GetNumInputs(const armnn::Layer& layer)
403{
404 return layer.GetNumInputSlots();
405}
406
407// Generic implementation to get the number of output slots for a given layer type;
408template<armnn::LayerType Type>
409unsigned int GetNumOutputs(const armnn::Layer& layer)
410{
411 return layer.GetNumOutputSlots();
412}
413
414template<>
415unsigned int GetNumInputs<armnn::LayerType::Merger>(const armnn::Layer& layer)
416{
417 boost::ignore_unused(layer);
418 return 2;
419}
420
telsoa01c577f2c2018-08-31 09:22:23 +0100421// Tests that the IsLayerSupported() function returns the correct value.
422// 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 +0000423// Returns true if expectations are met, otherwise returns false.
424template<typename FactoryType, armnn::DataType DataType, armnn::LayerType Type>
425bool IsLayerSupportedTest(FactoryType *factory, Tag<Type>)
426{
427 using LayerPolicy = LayerTypePolicy<Type, DataType>;
428 using LayerType = typename LayerPolicy::Type;
429 using LayerDesc = typename LayerPolicy::Desc;
430 DummyLayer<LayerType, LayerDesc> layer;
431
432 unsigned int numIn = GetNumInputs<Type>(*layer.m_Layer);
433 unsigned int numOut = GetNumOutputs<Type>(*layer.m_Layer);
434
telsoa01c577f2c2018-08-31 09:22:23 +0100435 // Make another dummy layer just to make IsLayerSupported have valid inputs.
telsoa014fcda012018-03-09 14:13:49 +0000436 DummyLayer<armnn::ConstantLayer, void> previousLayer;
telsoa01c577f2c2018-08-31 09:22:23 +0100437 // Set output of the previous layer to a dummy tensor.
telsoa014fcda012018-03-09 14:13:49 +0000438 armnn::TensorInfo output = MakeDummyTensorInfo<DataType>();
439 previousLayer.m_Layer->GetOutputSlot(0).SetTensorInfo(output);
telsoa01c577f2c2018-08-31 09:22:23 +0100440 // Connect all outputs of the previous layer to inputs of tested layer.
telsoa014fcda012018-03-09 14:13:49 +0000441 for (unsigned int i = 0; i < numIn; i++)
442 {
443 armnn::IOutputSlot& previousLayerOutputSlot = previousLayer.m_Layer->GetOutputSlot(0);
444 armnn::IInputSlot& layerInputSlot = layer.m_Layer->GetInputSlot(i);
445 previousLayerOutputSlot.Connect(layerInputSlot);
446 }
telsoa01c577f2c2018-08-31 09:22:23 +0100447 // Set outputs of tested layer to a dummy tensor.
telsoa014fcda012018-03-09 14:13:49 +0000448 for (unsigned int i = 0; i < numOut; i++)
449 {
450 layer.m_Layer->GetOutputSlot(0).SetTensorInfo(output);
451 }
452
453 std::string layerName = LayerPolicy::NameStr;
454 std::string reasonIfUnsupported;
455 if (FactoryType::IsLayerSupported(*layer.m_Layer, DataType, reasonIfUnsupported))
456 {
457 std::string errorMsg = " layer expected support but found none.";
458 try
459 {
460 bool retVal = LayerPolicy::MakeDummyWorkload(factory, numIn, numOut).get() != nullptr;
Matteo Martincighfbebcbd2018-10-16 09:45:08 +0100461 BOOST_CHECK_MESSAGE(retVal, layerName << errorMsg);
telsoa014fcda012018-03-09 14:13:49 +0000462 return retVal;
463 }
telsoa01c577f2c2018-08-31 09:22:23 +0100464 catch(const armnn::InvalidArgumentException& e)
telsoa014fcda012018-03-09 14:13:49 +0000465 {
466 boost::ignore_unused(e);
467 // This is ok since we throw InvalidArgumentException when creating the dummy workload.
468 return true;
469 }
470 catch(const std::exception& e)
471 {
472 errorMsg = e.what();
473 BOOST_TEST_ERROR(layerName << ": " << errorMsg);
474 return false;
475 }
telsoa01c577f2c2018-08-31 09:22:23 +0100476 catch(...)
telsoa014fcda012018-03-09 14:13:49 +0000477 {
478 errorMsg = "Unexpected error while testing support for ";
479 BOOST_TEST_ERROR(errorMsg << layerName);
480 return false;
481 }
482 }
483 else
484 {
485 std::string errorMsg = "layer expected no support (giving reason: " + reasonIfUnsupported + ") but found some.";
486 try
487 {
488 bool retVal = LayerPolicy::MakeDummyWorkload(factory, numIn, numOut).get() == nullptr;
489 BOOST_CHECK_MESSAGE(retVal, layerName << errorMsg);
490 return retVal;
491 }
492 // These two exceptions are ok: For workloads that are partially supported, attempting to instantiate them
493 // using parameters that make IsLayerSupported() return false should throw an
telsoa01c577f2c2018-08-31 09:22:23 +0100494 // InvalidArgumentException or UnimplementedException.
telsoa014fcda012018-03-09 14:13:49 +0000495 catch(const armnn::InvalidArgumentException& e)
496 {
497 boost::ignore_unused(e);
498 return true;
499 }
telsoa01c577f2c2018-08-31 09:22:23 +0100500 catch(const armnn::UnimplementedException& e)
telsoa014fcda012018-03-09 14:13:49 +0000501 {
502 boost::ignore_unused(e);
503 return true;
504 }
505 catch(const std::exception& e)
506 {
507 errorMsg = e.what();
508 BOOST_TEST_ERROR(layerName << ": " << errorMsg);
509 return false;
510 }
telsoa01c577f2c2018-08-31 09:22:23 +0100511 catch(...)
telsoa014fcda012018-03-09 14:13:49 +0000512 {
513 errorMsg = "Unexpected error while testing support for ";
514 BOOST_TEST_ERROR(errorMsg << layerName);
515 return false;
516 }
517 }
518}
519
telsoa01c577f2c2018-08-31 09:22:23 +0100520// Helper function to compute the next type in the LayerType enum.
telsoa014fcda012018-03-09 14:13:49 +0000521constexpr armnn::LayerType NextType(armnn::LayerType type)
522{
523 return static_cast<armnn::LayerType>(static_cast<int>(type)+1);
524}
525
telsoa01c577f2c2018-08-31 09:22:23 +0100526// Termination function for determining the end of the LayerType enumeration.
telsoa014fcda012018-03-09 14:13:49 +0000527template<typename FactoryType, armnn::DataType DataType, armnn::LayerType Type>
528bool IsLayerSupportedTestsImpl(FactoryType *factory, Tag<armnn::LayerType::LastLayer>)
529{
530 return IsLayerSupportedTest<FactoryType, DataType, Type>(factory, Tag<Type>());
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000531}
telsoa014fcda012018-03-09 14:13:49 +0000532
telsoa01c577f2c2018-08-31 09:22:23 +0100533// Recursive function to test and enter in the LayerType enum and then iterate on the next entry.
telsoa014fcda012018-03-09 14:13:49 +0000534template<typename FactoryType, armnn::DataType DataType, armnn::LayerType Type>
535bool IsLayerSupportedTestsImpl(FactoryType *factory, Tag<Type>)
536{
537 bool v = IsLayerSupportedTest<FactoryType, DataType, Type>(factory, Tag<Type>());
538
539 return v &&
540 IsLayerSupportedTestsImpl<FactoryType, DataType, NextType(Type)>
541 (factory, Tag<NextType(Type)>());
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000542}
telsoa014fcda012018-03-09 14:13:49 +0000543
544// Helper function to pass through to the test framework.
545template<typename FactoryType, armnn::DataType DataType>
546bool IsLayerSupportedTests(FactoryType *factory)
547{
548 return IsLayerSupportedTestsImpl<FactoryType, DataType>(factory, Tag<armnn::LayerType::FirstLayer>());
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000549}
telsoa014fcda012018-03-09 14:13:49 +0000550
551template<armnn::LayerType Type>
552bool TestLayerTypeMatches()
553{
554 using LayerPolicy = LayerTypePolicy<Type, armnn::DataType::Float32>;
555 using LayerType = typename LayerPolicy::Type;
556 using LayerDesc = typename LayerPolicy::Desc;
557 DummyLayer<LayerType, LayerDesc> layer;
558
559 std::stringstream ss;
560 ss << LayerPolicy::NameStr << " layer type mismatches expected layer type value.";
561 bool v = Type == layer.m_Layer->GetType();
562 BOOST_CHECK_MESSAGE(v, ss.str());
563 return v;
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000564}
telsoa014fcda012018-03-09 14:13:49 +0000565
566template<armnn::LayerType Type>
567bool LayerTypeMatchesTestImpl(Tag<armnn::LayerType::LastLayer>)
568{
569 return TestLayerTypeMatches<Type>();
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000570}
telsoa014fcda012018-03-09 14:13:49 +0000571
572template<armnn::LayerType Type>
573bool LayerTypeMatchesTestImpl(Tag<Type>)
574{
575 return TestLayerTypeMatches<Type>() &&
576 LayerTypeMatchesTestImpl<NextType(Type)>(Tag<NextType(Type)>());
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000577}
telsoa014fcda012018-03-09 14:13:49 +0000578
telsoa01c577f2c2018-08-31 09:22:23 +0100579template<typename FactoryType, typename LayerType, armnn::DataType InputDataType , armnn::DataType OutputDataType>
580bool IsConvertLayerSupportedTests(std::string& reasonIfUnsupported)
581{
582 armnn::Graph graph;
583 LayerType* const layer = graph.AddLayer<LayerType>("LayerName");
584
585 armnn::Layer* const input = graph.AddLayer<armnn::InputLayer>(0, "input");
586 armnn::Layer* const output = graph.AddLayer<armnn::OutputLayer>(0, "output");
587
588 armnn::TensorInfo inputTensorInfo({1, 3, 2, 3}, InputDataType);
589 armnn::TensorInfo outputTensorInfo({1, 3, 2, 3}, OutputDataType);
590
591 input->GetOutputSlot(0).Connect(layer->GetInputSlot(0));
592 input->GetOutputHandler(0).SetTensorInfo(inputTensorInfo);
593 layer->GetOutputSlot(0).Connect(output->GetInputSlot(0));
594 layer->GetOutputHandler(0).SetTensorInfo(outputTensorInfo);
595
596 bool result = FactoryType::IsLayerSupported(*layer, InputDataType, reasonIfUnsupported);
597
598 return result;
Matteo Martincigh59a950c2018-12-13 12:48:25 +0000599}
telsoa01c577f2c2018-08-31 09:22:23 +0100600
Matthew Bentham1f0ff352019-01-02 13:26:31 +0000601template<typename FactoryType, armnn::DataType InputDataType , armnn::DataType OutputDataType>
602bool IsMeanLayerSupportedTests(std::string& reasonIfUnsupported)
603{
604 armnn::Graph graph;
605 static const std::vector<unsigned> axes = {1, 0};
606 armnn::MeanDescriptor desc(axes, false);
607
608 armnn::Layer* const layer = graph.AddLayer<armnn::MeanLayer>(desc, "LayerName");
609
610 armnn::Layer* const input = graph.AddLayer<armnn::InputLayer>(0, "input");
611 armnn::Layer* const output = graph.AddLayer<armnn::OutputLayer>(0, "output");
612
613 armnn::TensorInfo inputTensorInfo({4, 3, 2}, InputDataType);
614 armnn::TensorInfo outputTensorInfo({2}, OutputDataType);
615
616 input->GetOutputSlot(0).Connect(layer->GetInputSlot(0));
617 input->GetOutputHandler(0).SetTensorInfo(inputTensorInfo);
618 layer->GetOutputSlot(0).Connect(output->GetInputSlot(0));
619 layer->GetOutputHandler(0).SetTensorInfo(outputTensorInfo);
620
621 bool result = FactoryType::IsLayerSupported(*layer, InputDataType, reasonIfUnsupported);
622
623 return result;
624}
625
626
telsoa014fcda012018-03-09 14:13:49 +0000627} //namespace