diff options
Diffstat (limited to 'src/backends/reference/test/RefLayerSupportTests.cpp')
-rw-r--r-- | src/backends/reference/test/RefLayerSupportTests.cpp | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefLayerSupportTests.cpp b/src/backends/reference/test/RefLayerSupportTests.cpp index 1d4b4a0f3a..2a27a9d20f 100644 --- a/src/backends/reference/test/RefLayerSupportTests.cpp +++ b/src/backends/reference/test/RefLayerSupportTests.cpp @@ -235,4 +235,41 @@ BOOST_AUTO_TEST_CASE(IsLayerNotSupportedMeanDimensionsReference) != std::string::npos); } +BOOST_AUTO_TEST_CASE(IsConstantSupportedRef) +{ + std::string reasonIfUnsupported; + + bool result = IsConstantLayerSupportedTests<armnn::RefWorkloadFactory, + armnn::DataType::Float16>(reasonIfUnsupported); + BOOST_CHECK(!result); + + result = IsConstantLayerSupportedTests<armnn::RefWorkloadFactory, + armnn::DataType::Float32>(reasonIfUnsupported); + BOOST_CHECK(result); + + result = IsConstantLayerSupportedTests<armnn::RefWorkloadFactory, + armnn::DataType::QAsymmU8>(reasonIfUnsupported); + BOOST_CHECK(result); + + result = IsConstantLayerSupportedTests<armnn::RefWorkloadFactory, + armnn::DataType::Boolean>(reasonIfUnsupported); + BOOST_CHECK(!result); + + result = IsConstantLayerSupportedTests<armnn::RefWorkloadFactory, + armnn::DataType::QSymmS16>(reasonIfUnsupported); + BOOST_CHECK(result); + + result = IsConstantLayerSupportedTests<armnn::RefWorkloadFactory, + armnn::DataType::QSymmS8>(reasonIfUnsupported); + BOOST_CHECK(result); + + result = IsConstantLayerSupportedTests<armnn::RefWorkloadFactory, + armnn::DataType::QAsymmS8>(reasonIfUnsupported); + BOOST_CHECK(result); + + result = IsConstantLayerSupportedTests<armnn::RefWorkloadFactory, + armnn::DataType::BFloat16>(reasonIfUnsupported); + BOOST_CHECK(result); +} + BOOST_AUTO_TEST_SUITE_END() |