diff options
Diffstat (limited to 'src/backends/backendsCommon/test')
-rw-r--r-- | src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp b/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp index dccfd1e75b..0780f4bd27 100644 --- a/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp +++ b/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp @@ -882,5 +882,22 @@ bool IsMeanLayerNotSupportedTests(std::string& reasonIfUnsupported) return result; } +template<typename FactoryType, armnn::DataType OutputDataType> +bool IsConstantLayerSupportedTests(std::string& reasonIfUnsupported) +{ + armnn::Graph graph; + + armnn::Layer* const layer = graph.AddLayer<armnn::ConstantLayer>("ConstantLayerName"); + armnn::Layer* const output = graph.AddLayer<armnn::OutputLayer>(0, "OutputLayerName"); + + armnn::TensorInfo outputTensorInfo({1, 1}, OutputDataType); + + layer->GetOutputSlot(0).Connect(output->GetInputSlot(0)); + layer->GetOutputHandler(0).SetTensorInfo(outputTensorInfo); + + bool result = FactoryType::IsLayerSupported(*layer, OutputDataType, reasonIfUnsupported); + + return result; +} } //namespace |