diff options
author | Mike Kelly <mike.kelly@arm.com> | 2020-04-27 09:55:40 +0100 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2020-04-27 10:19:33 +0000 |
commit | 0886ac4c403378ac0df85ab520a5db319be525a4 (patch) | |
tree | d21097f081c9879750f577b4760b7c26e83affb7 /src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp | |
parent | d7fcafaaaa917db458be362060e1b7855cad8083 (diff) | |
download | armnn-0886ac4c403378ac0df85ab520a5db319be525a4.tar.gz |
MLCE-190: Neon and CL Constant Workloads do not support newer DataTypes
* Added support for QASYMM8_SIGNED, QSYMM16, QSYMM8 and QSYMM8_PER_CHANNEL to Neon and CL backends
* Added unit tests to Neon, CL and Ref backends
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I4c726b6d86b4d75abedd130dcea372d1e82be5c2
Diffstat (limited to 'src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp')
-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 |