diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2020-07-27 11:27:19 +0100 |
|---|---|---|
| committer | Jim Flynn <jim.flynn@arm.com> | 2020-07-28 14:45:10 +0000 |
| commit | 4408850bd426481e39ae9182aff4d563b6836181 (patch) | |
| tree | 2fa8ad2f7ed34c249a6f9b701ad2cd0caa91a389 /src/backends/reference | |
| parent | c135179c935e3f85e591014e14be81b3f2597825 (diff) | |
| download | armnn-4408850bd426481e39ae9182aff4d563b6836181.tar.gz | |
IVGCVSW-4712 Add EndToEnd test for FILL
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ic89bcbbd580abe1b05bd26748db704e83cf65bea
Diffstat (limited to 'src/backends/reference')
| -rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 2 | ||||
| -rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 16 |
2 files changed, 18 insertions, 0 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index 877d200208..1886d3253b 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -877,6 +877,8 @@ bool RefLayerSupport::IsFillSupported(const TensorInfo& input, supported &= CheckSupportRule(TypeAnyOf(input, supportedTypes), reasonIfUnsupported, "Reference Fill: input type not supported."); + supported &= CheckSupportRule(TypeAnyOf(output, supportedTypes), reasonIfUnsupported, + "Reference Fill: output type not supported."); return supported; } diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index 45605e0fc6..88876877bf 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -14,6 +14,7 @@ #include <backendsCommon/test/DequantizeEndToEndTestImpl.hpp> #include <backendsCommon/test/DetectionPostProcessEndToEndTestImpl.hpp> #include <backendsCommon/test/ElementwiseUnaryEndToEndTestImpl.hpp> +#include <backendsCommon/test/FillEndToEndTestImpl.hpp> #include <backendsCommon/test/GatherEndToEndTestImpl.hpp> #include <backendsCommon/test/InstanceNormalizationEndToEndTestImpl.hpp> #include <backendsCommon/test/LogSoftmaxEndToEndTestImpl.hpp> @@ -582,6 +583,21 @@ BOOST_AUTO_TEST_CASE(RefEluEndToEndTestQSymmS16) EluEndToEndTest<armnn::DataType::QSymmS16>(defaultBackends); } +BOOST_AUTO_TEST_CASE(RefFillEndToEndTest) +{ + FillEndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +//BOOST_AUTO_TEST_CASE(RefFillEndToEndTestFloat16) +//{ +// FillEndToEnd<armnn::DataType::Float16>(defaultBackends); +//} + +BOOST_AUTO_TEST_CASE(RefFillEndToEndTestInt32) +{ + FillEndToEnd<armnn::DataType::Signed32>(defaultBackends); +} + BOOST_AUTO_TEST_CASE(RefGatherFloatTest) { GatherEndToEnd<armnn::DataType::Float32>(defaultBackends); |