diff options
Diffstat (limited to 'src/backends/reference/test/RefEndToEndTests.cpp')
| -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) |