aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/test/ClEndToEndTests.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/cl/test/ClEndToEndTests.cpp')
-rw-r--r--src/backends/cl/test/ClEndToEndTests.cpp15
1 files changed, 15 insertions, 0 deletions
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);
+}
+
}