diff options
Diffstat (limited to 'src/backends/reference')
-rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 2 | ||||
-rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 8 |
2 files changed, 5 insertions, 5 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index 80527bdcb2..f6c214e2c6 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -874,7 +874,7 @@ bool RefLayerSupport::IsFillSupported(const TensorInfo& input, DataType::Signed32 }; - supported &= CheckSupportRule(TypeAnyOf(input, supportedTypes), reasonIfUnsupported, + supported &= CheckSupportRule(TypeIs(input, DataType::Signed32), reasonIfUnsupported, "Reference Fill: input type not supported."); supported &= CheckSupportRule(TypeAnyOf(output, supportedTypes), reasonIfUnsupported, diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index 027b621c23..4598568070 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -589,10 +589,10 @@ BOOST_AUTO_TEST_CASE(RefFillEndToEndTest) FillEndToEnd<armnn::DataType::Float32>(defaultBackends); } -//BOOST_AUTO_TEST_CASE(RefFillEndToEndTestFloat16) -//{ -// FillEndToEnd<armnn::DataType::Float16>(defaultBackends); -//} +BOOST_AUTO_TEST_CASE(RefFillEndToEndTestFloat16) +{ + FillEndToEnd<armnn::DataType::Float16>(defaultBackends); +} BOOST_AUTO_TEST_CASE(RefFillEndToEndTestInt32) { |