diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-12-15 14:20:47 +0000 |
---|---|---|
committer | john.mcloughlin <john.mcloughlin@arm.com> | 2023-12-21 11:14:54 +0000 |
commit | ca5a23a7cbe46b8da8de432d80889c47a745ca4c (patch) | |
tree | c375efe1f4ecdad708d6e8e771a13c07ddbc0257 /src/backends/reference/test/RefEndToEndTests.cpp | |
parent | 0587dd01272199a36210bd0ccc266185b113df75 (diff) | |
download | armnn-ca5a23a7cbe46b8da8de432d80889c47a745ca4c.tar.gz |
Add Quantize Support to TOSA Ref Backend
* Adding a one to many tosa mapping for Quantize
* Added tests
* Resolves IVGCVSW-7175
Signed-off-by: John Mcloughlin <john.mcloughlin@arm.com>
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ia0852fefb618b4a29c2601b9de8b6b2731229801
Diffstat (limited to 'src/backends/reference/test/RefEndToEndTests.cpp')
-rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 49 |
1 files changed, 41 insertions, 8 deletions
diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index 8de763e07e..143d85ac9b 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -29,6 +29,7 @@ #include <backendsCommon/test/LogSoftmaxEndToEndTestImpl.hpp> #include <backendsCommon/test/PreluEndToEndTestImpl.hpp> #include <backendsCommon/test/QLstmEndToEndTestImpl.hpp> +#include <backendsCommon/test/QuantizationEndToEndTestImpl.hpp> #include <backendsCommon/test/RankEndToEndTestImpl.hpp> #include <backendsCommon/test/ReduceEndToEndTestImpl.hpp> #include <backendsCommon/test/ReshapeEndToEndTestImpl.hpp> @@ -792,42 +793,42 @@ TEST_CASE("RefGatherNdMultiDimInt16Test") } // DepthToSpace -TEST_CASE("DephtToSpaceEndToEndNchwFloat32") +TEST_CASE("DepthToSpaceEndToEndNchwFloat32") { DepthToSpaceEndToEnd<armnn::DataType::Float32>(defaultBackends, armnn::DataLayout::NCHW); } -TEST_CASE("DephtToSpaceEndToEndNchwFloat16") +TEST_CASE("DepthToSpaceEndToEndNchwFloat16") { DepthToSpaceEndToEnd<armnn::DataType::Float16>(defaultBackends, armnn::DataLayout::NCHW); } -TEST_CASE("DephtToSpaceEndToEndNchwUint8") +TEST_CASE("DepthToSpaceEndToEndNchwUint8") { DepthToSpaceEndToEnd<armnn::DataType::QAsymmU8>(defaultBackends, armnn::DataLayout::NCHW); } -TEST_CASE("DephtToSpaceEndToEndNchwInt16") +TEST_CASE("DepthToSpaceEndToEndNchwInt16") { DepthToSpaceEndToEnd<armnn::DataType::QSymmS16>(defaultBackends, armnn::DataLayout::NCHW); } -TEST_CASE("DephtToSpaceEndToEndNhwcFloat32") +TEST_CASE("DepthToSpaceEndToEndNhwcFloat32") { DepthToSpaceEndToEnd<armnn::DataType::Float32>(defaultBackends, armnn::DataLayout::NHWC); } -TEST_CASE("DephtToSpaceEndToEndNhwcFloat16") +TEST_CASE("DepthToSpaceEndToEndNhwcFloat16") { DepthToSpaceEndToEnd<armnn::DataType::Float16>(defaultBackends, armnn::DataLayout::NHWC); } -TEST_CASE("DephtToSpaceEndToEndNhwcUint8") +TEST_CASE("DepthToSpaceEndToEndNhwcUint8") { DepthToSpaceEndToEnd<armnn::DataType::QAsymmU8>(defaultBackends, armnn::DataLayout::NHWC); } -TEST_CASE("DephtToSpaceEndToEndNhwcInt16") +TEST_CASE("DepthToSpaceEndToEndNhwcInt16") { DepthToSpaceEndToEnd<armnn::DataType::QSymmS16>(defaultBackends, armnn::DataLayout::NHWC); } @@ -1064,6 +1065,38 @@ TEST_CASE("RefPreluEndToEndTestQSymm16") PreluEndToEndPositiveTest<armnn::DataType::QSymmS16>(defaultBackends); } +// Quantization +TEST_CASE("QuantizationEndToEndFloat32_U8Test") +{ + QuantizationEndToEndFloat32<armnn::DataType::QAsymmU8>(defaultBackends); +} + +TEST_CASE("QuantizationEndToEndFloat32_I8Test") +{ + QuantizationEndToEndFloat32<armnn::DataType::QAsymmS8>(defaultBackends); +} + +TEST_CASE("QuantizationEndToEndFloat32_S16Test") +{ + QuantizationEndToEndFloat32<armnn::DataType::QSymmS16>(defaultBackends); +} + +TEST_CASE("QuantizationEndToEndFloat16_U8Test") +{ + QuantizationEndToEndFloat16<armnn::DataType::QAsymmU8>(defaultBackends); +} + +TEST_CASE("QuantizationEndToEndFloat16_I8Test") +{ + QuantizationEndToEndFloat16<armnn::DataType::QAsymmS8>(defaultBackends); +} + +TEST_CASE("QuantizationEndToEndFloat16_S16Test") +{ + QuantizationEndToEndFloat16<armnn::DataType::QSymmS16>(defaultBackends); +} + +// SpaceToDepth TEST_CASE("RefSpaceToDepthNhwcEndToEndTest1") { SpaceToDepthNhwcEndToEndTest1(defaultBackends); |