From 1bf56cde4f199ea8df722d1875ee44bd0ffb414a Mon Sep 17 00:00:00 2001
From: Declan-ARM <decmce01@arm.com>
Date: Thu, 20 Jul 2023 17:32:57 +0100
Subject: IVGCVSW-7836 Add ReverseV2 End-to-End Tests

  * create implementation header
  * add copyright notice
  * add pragma once and anonymous namespace
  * create network function declaration
  * complete body of network function
  * create end-to-end function declaration
  * complete body of end-to-end function
  * add references to tests for supported data types

Signed-off-by: Declan-ARM <decmce01@arm.com>
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I376e48efd8b6ca9e0e0b05b516be599c0acdbd16
---
 src/backends/reference/RefLayerSupport.cpp       |  6 ++--
 src/backends/reference/test/RefEndToEndTests.cpp | 37 ++++++++++++++++++++++++
 2 files changed, 41 insertions(+), 2 deletions(-)

(limited to 'src/backends/reference')

diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp
index 9d396e5db9..2e845a8c51 100644
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -2373,14 +2373,16 @@ bool RefLayerSupport::IsReverseV2Supported(const TensorInfo& input0,
 {
     bool supported = true;
     // ReverseV2 is data type agnostic so it can support all the types in the Reference backend
-    std::array<DataType,6> supportedTypes =
+    std::array<DataType,8> supportedTypes =
     {
         DataType::BFloat16,
         DataType::Float32,
         DataType::Float16,
         DataType::QAsymmS8,
         DataType::QAsymmU8,
-        DataType::QSymmS16
+        DataType::QSymmS8,
+        DataType::QSymmS16,
+        DataType::Signed32
     };
 
     supported &= CheckSupportRule(TypeAnyOf(input0, supportedTypes), reasonIfUnsupported,
diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp
index 6f05ba77cb..fc62a89dd7 100644
--- a/src/backends/reference/test/RefEndToEndTests.cpp
+++ b/src/backends/reference/test/RefEndToEndTests.cpp
@@ -32,6 +32,7 @@
 #include <backendsCommon/test/ReduceEndToEndTestImpl.hpp>
 #include <backendsCommon/test/ReshapeEndToEndTestImpl.hpp>
 #include <backendsCommon/test/ResizeEndToEndTestImpl.hpp>
+#include <backendsCommon/test/ReverseV2EndToEndTestImpl.hpp>
 #include <backendsCommon/test/SpaceToDepthEndToEndTestImpl.hpp>
 #include <backendsCommon/test/SplitterEndToEndTestImpl.hpp>
 #include <backendsCommon/test/StridedSliceAsyncEndToEndTest.hpp>
@@ -1322,6 +1323,42 @@ TEST_CASE("RefResizeNearestNeighborEndToEndInt16NhwcTest")
     ResizeNearestNeighborEndToEnd<armnn::DataType::QSymmS16>(defaultBackends, armnn::DataLayout::NHWC);
 }
 
+// ReverseV2
+TEST_CASE("RefReverseV2EndToEndFloat16Test")
+{
+    ReverseV2EndToEnd<armnn::DataType::Float16>(defaultBackends);
+}
+
+TEST_CASE("RefReverseV2EndToEndFloat32Test")
+{
+    ReverseV2EndToEnd<armnn::DataType::Float32>(defaultBackends);
+}
+
+TEST_CASE("RefReverseV2EndToEndQAsymmS8Test")
+{
+    ReverseV2EndToEnd<armnn::DataType::QAsymmS8>(defaultBackends);
+}
+
+TEST_CASE("RefReverseV2EndToEndQAsymmU8Test")
+{
+    ReverseV2EndToEnd<armnn::DataType::QAsymmU8>(defaultBackends);
+}
+
+TEST_CASE("RefReverseV2EndToEndQSymmS8Test")
+{
+    ReverseV2EndToEnd<armnn::DataType::QSymmS8>(defaultBackends);
+}
+
+TEST_CASE("RefReverseV2EndToEndQSymmS16Test")
+{
+    ReverseV2EndToEnd<armnn::DataType::QSymmS16>(defaultBackends);
+}
+
+TEST_CASE("RefReverseV2EndToEndSigned32Test")
+{
+    ReverseV2EndToEnd<armnn::DataType::Signed32>(defaultBackends);
+}
+
 // InstanceNormalization
 TEST_CASE("RefInstanceNormalizationNhwcEndToEndTest1")
 {
-- 
cgit v1.2.1