diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2020-07-29 14:29:20 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-07-30 13:25:34 +0000 |
commit | 2e3f4d2330094b4c8afd9872c2cf6765d69e5bdb (patch) | |
tree | ac0e683334058a67c0dbf9a66ec561f89381d46b /src/backends/reference | |
parent | ef2838bbf3b64e045538e94493acc596da184143 (diff) | |
download | armnn-2e3f4d2330094b4c8afd9872c2cf6765d69e5bdb.tar.gz |
IVGCVSW-4713 Add EndToEnd test for RANK
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ia8f237a2500986e01843defb75787694a20ba24c
Diffstat (limited to 'src/backends/reference')
-rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index 88876877bf..027b621c23 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -20,6 +20,7 @@ #include <backendsCommon/test/LogSoftmaxEndToEndTestImpl.hpp> #include <backendsCommon/test/PreluEndToEndTestImpl.hpp> #include <backendsCommon/test/QLstmEndToEndTestImpl.hpp> +#include <backendsCommon/test/RankEndToEndTestImpl.hpp> #include <backendsCommon/test/ResizeEndToEndTestImpl.hpp> #include <backendsCommon/test/SpaceToDepthEndToEndTestImpl.hpp> #include <backendsCommon/test/SplitterEndToEndTestImpl.hpp> @@ -1257,6 +1258,36 @@ BOOST_AUTO_TEST_CASE(RefQLstmEndToEndTest) QLstmEndToEnd(defaultBackends); } +BOOST_AUTO_TEST_CASE(RefRankEndToEndTest) +{ + RankEndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefRankEndToEndTestFloat16) +{ + RankEndToEnd<armnn::DataType::Float16>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefRankEndToEndTestInt32) +{ + RankEndToEnd<armnn::DataType::Signed32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefRankEndToEndTestQAsymmS8) +{ + RankEndToEnd<armnn::DataType::QAsymmS8>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefRankEndToEndTestQSymmS16) +{ + RankEndToEnd<armnn::DataType::QSymmS16>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefRankEndToEndTestQSymmS8) +{ + RankEndToEnd<armnn::DataType::QSymmS8>(defaultBackends); +} + #if !defined(__ANDROID__) // Only run these tests on non Android platforms BOOST_AUTO_TEST_CASE(RefImportNonAlignedPointerTest) |