From 2e3f4d2330094b4c8afd9872c2cf6765d69e5bdb Mon Sep 17 00:00:00 2001
From: Teresa Charlin <teresa.charlinreyes@arm.com>
Date: Wed, 29 Jul 2020 14:29:20 +0100
Subject: IVGCVSW-4713 Add EndToEnd test for RANK

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ia8f237a2500986e01843defb75787694a20ba24c
---
 src/backends/reference/test/RefEndToEndTests.cpp | 31 ++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

(limited to 'src/backends/reference/test')

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)
-- 
cgit v1.2.1