diff options
| -rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 3 | ||||
| -rw-r--r-- | src/backends/reference/test/RefLayerSupportTests.cpp | 2 |
2 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index 034cd12477..3bcb7e0261 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -366,9 +366,10 @@ bool RefLayerSupport::IsConcatSupported(const std::vector<const TensorInfo*> inp bool RefLayerSupport::IsConstantSupported(const TensorInfo& output, Optional<std::string&> reasonIfUnsupported) const { - std::array<DataType,7> supportedTypes = + std::array<DataType,8> supportedTypes = { DataType::BFloat16, + DataType::Float16, DataType::Float32, DataType::QAsymmS8, DataType::QAsymmU8, diff --git a/src/backends/reference/test/RefLayerSupportTests.cpp b/src/backends/reference/test/RefLayerSupportTests.cpp index 2a27a9d20f..1648583210 100644 --- a/src/backends/reference/test/RefLayerSupportTests.cpp +++ b/src/backends/reference/test/RefLayerSupportTests.cpp @@ -241,7 +241,7 @@ BOOST_AUTO_TEST_CASE(IsConstantSupportedRef) bool result = IsConstantLayerSupportedTests<armnn::RefWorkloadFactory, armnn::DataType::Float16>(reasonIfUnsupported); - BOOST_CHECK(!result); + BOOST_CHECK(result); result = IsConstantLayerSupportedTests<armnn::RefWorkloadFactory, armnn::DataType::Float32>(reasonIfUnsupported); |