diff options
Diffstat (limited to 'src/backends/neon/test/NeonLayerSupportTests.cpp')
-rw-r--r-- | src/backends/neon/test/NeonLayerSupportTests.cpp | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/src/backends/neon/test/NeonLayerSupportTests.cpp b/src/backends/neon/test/NeonLayerSupportTests.cpp index 2d43125506..3b086ad28f 100644 --- a/src/backends/neon/test/NeonLayerSupportTests.cpp +++ b/src/backends/neon/test/NeonLayerSupportTests.cpp @@ -85,4 +85,41 @@ BOOST_AUTO_TEST_CASE(IsMeanSupportedNeon) BOOST_CHECK(result); } +BOOST_AUTO_TEST_CASE(IsConstantSupportedNeon) +{ + std::string reasonIfUnsupported; + + bool result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory, + armnn::DataType::Float16>(reasonIfUnsupported); + BOOST_CHECK(result); + + result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory, + armnn::DataType::Float32>(reasonIfUnsupported); + BOOST_CHECK(result); + + result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory, + armnn::DataType::QAsymmU8>(reasonIfUnsupported); + BOOST_CHECK(result); + + result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory, + armnn::DataType::Boolean>(reasonIfUnsupported); + BOOST_CHECK(!result); + + result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory, + armnn::DataType::QSymmS16>(reasonIfUnsupported); + BOOST_CHECK(result); + + result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory, + armnn::DataType::QSymmS8>(reasonIfUnsupported); + BOOST_CHECK(result); + + result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory, + armnn::DataType::QAsymmS8>(reasonIfUnsupported); + BOOST_CHECK(result); + + result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory, + armnn::DataType::BFloat16>(reasonIfUnsupported); + BOOST_CHECK(result); +} + BOOST_AUTO_TEST_SUITE_END() |