diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2024-03-05 17:59:27 +0000 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2024-03-12 16:57:28 +0000 |
| commit | 70dc5e9dc375c2feb306c7e709078c9f61ef8b1a (patch) | |
| tree | f85bea5ae8fdffe513d44edfc616d8a0eb690936 /src/backends/reference/test/RefEndToEndTests.cpp | |
| parent | ce7f51f0a8f815ff0e0dd94e209a8cf1802ac914 (diff) | |
| download | armnn-70dc5e9dc375c2feb306c7e709078c9f61ef8b1a.tar.gz | |
IVGCVSW-8233 ScatterNd End to End tests added
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Id89233954dd8da600c2f82e718df849b098c8af4
Diffstat (limited to 'src/backends/reference/test/RefEndToEndTests.cpp')
| -rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index 4af54852fc..73786b5ccd 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -37,6 +37,7 @@ #include <backendsCommon/test/ReshapeEndToEndTestImpl.hpp> #include <backendsCommon/test/ResizeEndToEndTestImpl.hpp> #include <backendsCommon/test/ReverseV2EndToEndTestImpl.hpp> +#include <backendsCommon/test/ScatterNdEndToEndTestImpl.hpp> #include <backendsCommon/test/SliceEndToEndTestImpl.hpp> #include <backendsCommon/test/SoftmaxEndToEndTestImpl.hpp> #include <backendsCommon/test/SpaceToDepthEndToEndTestImpl.hpp> @@ -1339,6 +1340,49 @@ TEST_CASE("QuantizationEndToEndFloat16_S16Test") QuantizationEndToEndFloat16<armnn::DataType::QSymmS16>(defaultBackends); } +// ScatterNd + +TEST_CASE("RefScatterNd1DInputEndToEndFloat32Test") +{ + ScatterNd1DimUpdateWithInputEndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +TEST_CASE("RefScatterNd1DInputEndToEndInt8Test") +{ + ScatterNd1DimUpdateWithInputEndToEnd<armnn::DataType::QAsymmS8>(defaultBackends); +} + +TEST_CASE("RefScatterNd1DNoInputEndToEndFloat32Test") +{ + ScatterNd1DimUpdateNoInputEndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +TEST_CASE("RefScatterNd1DNoInputEndToEndInt8Test") +{ + ScatterNd1DimUpdateNoInputEndToEnd<armnn::DataType::QAsymmS8>(defaultBackends); +} + +TEST_CASE("RefScatterNd2DInputEndToEndFloat32Test") +{ + ScatterNd2DimUpdateWithInputEndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +TEST_CASE("RefScatterNd2DInputEndToEndInt8Test") +{ + ScatterNd2DimUpdateWithInputEndToEnd<armnn::DataType::QAsymmS8>(defaultBackends); +} + +TEST_CASE("RefScatterNd2DNoInputEndToEndFloat32Test") +{ + ScatterNd2DimUpdateNoInputEndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +TEST_CASE("RefScatterNd2DNoInputEndToEndInt8Test") +{ + ScatterNd2DimUpdateNoInputEndToEnd<armnn::DataType::QAsymmS8>(defaultBackends); +} + + // SpaceToDepth TEST_CASE("RefSpaceToDepthNhwcEndToEndTest1") { |