diff options
Diffstat (limited to 'src/backends/backendsCommon/test')
| -rw-r--r-- | src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp b/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp index c52d6a9511..521eef6230 100644 --- a/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp +++ b/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp @@ -362,6 +362,7 @@ struct LayerTypePolicy<armnn::LayerType::name, DataType> \ using Desc = descType; \ using QueueDesc = armnn::name##QueueDescriptor; \ constexpr static const char* NameStr = #name; \ + constexpr static const bool IsException = true; \ \ static std::unique_ptr<armnn::IWorkload> MakeDummyWorkload(armnn::IWorkloadFactory *factory, \ unsigned int nIn, unsigned int nOut) \ @@ -388,6 +389,7 @@ struct LayerTypePolicy<armnn::LayerType::name, DataType> \ using Type = armnn::name##Layer; \ using Desc = descType; \ constexpr static const char* NameStr = #name; \ + constexpr static const bool IsException = true; \ \ static std::unique_ptr<armnn::IWorkload> MakeDummyWorkload(armnn::IWorkloadFactory *factory, \ unsigned int nIn, unsigned int nOut) \ @@ -551,6 +553,11 @@ bool IsLayerSupportedTest(FactoryType *factory, Tag<Type>) using LayerDesc = typename LayerPolicy::Desc; DummyLayer<LayerType, LayerDesc> layer; + if (LayerPolicy::IsException) //Don't test exceptions to the rule. + { + return true; + } + unsigned int numIn = GetNumInputs<Type>(*layer.m_Layer); unsigned int numOut = GetNumOutputs<Type>(*layer.m_Layer); |