From a7bd3fa77431f0e99698f24e325b85df459d3b10 Mon Sep 17 00:00:00 2001
From: Mike Kelly <mike.kelly@arm.com>
Date: Wed, 20 Dec 2023 12:48:02 +0000
Subject: IVGCVSW-7830 Remove Reshape where possible

 * Remove reshape on ClBackend
 * Remove unnecessary restriction on NeonBackend remove Reshape

Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I79940c9f8609d19b79f2fbe55225ffc8f0d90c25
---
 src/backends/cl/test/ClEndToEndTests.cpp | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

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

diff --git a/src/backends/cl/test/ClEndToEndTests.cpp b/src/backends/cl/test/ClEndToEndTests.cpp
index 2436a8223f..78d2dea90d 100644
--- a/src/backends/cl/test/ClEndToEndTests.cpp
+++ b/src/backends/cl/test/ClEndToEndTests.cpp
@@ -673,4 +673,19 @@ TEST_CASE("ClForceImportWithMisalignedInputAndOutputBuffersEndToEndTest")
     ForceImportWithMisalignedInputAndOutputBuffersEndToEndTest(clDefaultBackends);
 }
 
+TEST_CASE("ClReshapeRemovalSimpleCaseEndToEnd")
+{
+    ReshapeRemovalEndToEnd<armnn::DataType::Float32>(clDefaultBackends);
+}
+
+TEST_CASE("ClReshapeRemovalNCHWFirstEndToEnd")
+{
+    ReshapeRemovalNCHWEndToEnd<armnn::DataType::Float32>(clDefaultBackends, false, true);
+}
+
+TEST_CASE("ClReshapeRemovalNCHWSecondEndToEnd")
+{
+    ReshapeRemovalNCHWEndToEnd<armnn::DataType::Float32>(clDefaultBackends, false, false);
+}
+
 }
-- 
cgit v1.2.1