diff options
Diffstat (limited to 'src/backends/cl/test/ClEndToEndTests.cpp')
-rw-r--r-- | src/backends/cl/test/ClEndToEndTests.cpp | 15 |
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); +} + } |