diff options
Diffstat (limited to 'src/backends/neon')
-rw-r--r-- | src/backends/neon/NeonLayerSupport.cpp | 14 | ||||
-rw-r--r-- | src/backends/neon/test/NeonEndToEndTests.cpp | 5 |
2 files changed, 14 insertions, 5 deletions
diff --git a/src/backends/neon/NeonLayerSupport.cpp b/src/backends/neon/NeonLayerSupport.cpp index b848f0c10f..9dc8a01778 100644 --- a/src/backends/neon/NeonLayerSupport.cpp +++ b/src/backends/neon/NeonLayerSupport.cpp @@ -406,12 +406,16 @@ bool NeonLayerSupport::IsElementwiseUnarySupported(const TensorInfo& input, } } -bool NeonLayerSupport::IsFillSupported(const TensorInfo&, - const TensorInfo&, - const FillDescriptor&, - Optional<std::string&>) const +bool NeonLayerSupport::IsFillSupported(const TensorInfo& input, + const TensorInfo& output, + const FillDescriptor& descriptor, + Optional<std::string&> reasonIfUnsupported) const { - return true; + armnn::IgnoreUnused(input); + armnn::IgnoreUnused(output); + armnn::IgnoreUnused(descriptor); + + return IsNeonBackendSupported(reasonIfUnsupported); } bool NeonLayerSupport::IsFloorSupported(const TensorInfo& input, diff --git a/src/backends/neon/test/NeonEndToEndTests.cpp b/src/backends/neon/test/NeonEndToEndTests.cpp index e89718a05e..ffbae51df6 100644 --- a/src/backends/neon/test/NeonEndToEndTests.cpp +++ b/src/backends/neon/test/NeonEndToEndTests.cpp @@ -452,6 +452,11 @@ BOOST_AUTO_TEST_CASE(NeonFillEndToEndTest) FillEndToEnd<armnn::DataType::Float32>(defaultBackends); } +BOOST_AUTO_TEST_CASE(RefFillEndToEndTestFloat16) +{ + FillEndToEnd<armnn::DataType::Float16>(defaultBackends); +} + BOOST_AUTO_TEST_CASE(NeonFillEndToEndTestInt32) { FillEndToEnd<armnn::DataType::Signed32>(defaultBackends); |