diff options
author | Kevin May <kevin.may@arm.com> | 2023-12-12 11:18:46 +0000 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2023-12-14 10:05:56 +0000 |
commit | 1bea6beb042635c7716ae43220ee19eedb2de9ff (patch) | |
tree | 638a60e9128e28efb613bee96e2de62f2c08b3fc /src/backends/reference | |
parent | ce65588484ed1e553bdebf24123a30b5575f1bce (diff) | |
download | armnn-1bea6beb042635c7716ae43220ee19eedb2de9ff.tar.gz |
Add Split support to TOSA Reference Backend
* Resolves IVGCVSW-7918
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: Ic2afaa55f7ee88ce4c9b8ea696eef5f28663f8c6
Diffstat (limited to 'src/backends/reference')
-rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index 7e07c658fa..9fb6227663 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -1084,6 +1084,17 @@ TEST_CASE("RefSpaceToDepthNchwEndToEndTest2") SpaceToDepthNchwEndToEndTest2(defaultBackends); } +// Split +TEST_CASE("RefSplit1dEndtoEndTestSigned16") +{ + Splitter1dEndToEnd<DataType::QSymmS16>(defaultBackends); +} + +TEST_CASE("TosaRefSplit1dEndtoEndTestFloat16") +{ + Splitter1dEndToEndFloat16<DataType::Float16>(defaultBackends); +} + TEST_CASE("RefSplitter1dEndToEndTest") { Splitter1dEndToEnd<armnn::DataType::Float32>(defaultBackends); @@ -1114,6 +1125,16 @@ TEST_CASE("RefSplitter2dDim1EndToEndUint8Test") Splitter2dDim1EndToEnd<armnn::DataType::QAsymmU8>(defaultBackends); } +TEST_CASE("RefSplit3dDim1EndtoEndTestSigned16") +{ + Splitter3dDim1EndToEnd<DataType::QSymmS16>(defaultBackends); +} + +TEST_CASE("RefSplit3dDim1EndtoEndTestFloat16") +{ + Splitter3dDim1EndToEndFloat16<DataType::Float16>(defaultBackends); +} + TEST_CASE("RefSplitter3dDim0EndToEndTest") { Splitter3dDim0EndToEnd<armnn::DataType::Float32>(defaultBackends); @@ -1159,6 +1180,11 @@ TEST_CASE("RefSplitter4dDim2EndToEndTest") Splitter4dDim2EndToEnd<armnn::DataType::Float32>(defaultBackends); } +TEST_CASE("RefSplit4dDim2EndtoEndTestFloat16") +{ + Splitter4dDim2EndToEndFloat16<DataType::Float16>(defaultBackends); +} + TEST_CASE("RefSplitter4dDim3EndToEndTest") { Splitter4dDim3EndToEnd<armnn::DataType::Float32>(defaultBackends); |