diff options
Diffstat (limited to 'src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp b/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp index ed4b6ff09a..1078c2a8d7 100644 --- a/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp +++ b/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp @@ -6,6 +6,7 @@ #include <Graph.hpp> +#include <backendsCommon/MapWorkload.hpp> #include <backendsCommon/WorkloadFactory.hpp> #include <armnn/utility/IgnoreUnused.hpp> @@ -165,6 +166,22 @@ struct DummyLayer<armnn::ConcatLayer> }; template<> +struct DummyLayer<armnn::MapLayer, void> +{ + DummyLayer() + { + m_Layer = dummyGraph.AddLayer<armnn::MapLayer>(""); + } + + ~DummyLayer() + { + dummyGraph.EraseLayer(m_Layer); + } + + armnn::MapLayer* m_Layer; +}; + +template<> struct DummyLayer<armnn::OutputLayer, armnn::LayerBindingId> { DummyLayer() @@ -467,6 +484,27 @@ struct LayerTypePolicy<armnn::LayerType::name, DataType> \ } \ }; +#define DECLARE_LAYER_POLICY_MAP_PARAM(name, descType) \ +template<armnn::DataType DataType> \ +struct LayerTypePolicy<armnn::LayerType::name, DataType> \ +{ \ + using Type = armnn::name##Layer; \ + using Desc = descType; \ + using QueueDesc = armnn::name##QueueDescriptor; \ + using Workload = armnn::name##Workload; \ + constexpr static const char* NameStr = #name; \ + constexpr static const bool IsException = false; \ + \ + static std::unique_ptr<armnn::IWorkload> MakeDummyWorkload(armnn::IWorkloadFactory* factory, \ + unsigned int nIn, unsigned int nOut) \ + { \ + IgnoreUnused(factory); \ + QueueDesc desc; \ + armnn::WorkloadInfo info = MakeDummyWorkloadInfo<DataType>(nIn, nOut); \ + return std::make_unique<armnn::name##Workload>(desc, info); \ + } \ +}; + // Define a layer policy specialization for use with the IsLayerSupported tests. // Use this version for layers whose constructor takes 1 parameter(name). #define DECLARE_LAYER_POLICY_1_PARAM(name) DECLARE_LAYER_POLICY_CUSTOM_PARAM(name, void) @@ -563,6 +601,8 @@ DECLARE_LAYER_POLICY_2_PARAM(LogSoftmax) DECLARE_LAYER_POLICY_2_PARAM(Lstm) +DECLARE_LAYER_POLICY_MAP_PARAM(Map, void) + DECLARE_LAYER_POLICY_1_PARAM(Maximum) DECLARE_LAYER_POLICY_2_PARAM(Mean) @@ -751,6 +791,13 @@ bool IsLayerSupportedTest(FactoryType *factory, Tag<Type>) } } +template<typename FactoryType, armnn::DataType DataType, armnn::LayerType Type> +bool IsLayerSupportedTest(FactoryType *factory, Tag<armnn::LayerType::Map>) +{ + IgnoreUnused(factory); + return true; +} + // Helper function to compute the next type in the LayerType enum. constexpr armnn::LayerType NextType(armnn::LayerType type) { |