diff options
| author | James Conroy <james.conroy@arm.com> | 2019-11-13 15:35:59 +0000 |
|---|---|---|
| committer | James Conroy <james.conroy@arm.com> | 2019-11-13 16:13:37 +0000 |
| commit | 68a4dbcebbf3c8e759c3f809efbaee2560dd6381 (patch) | |
| tree | 822701de04e760dc8b2837b654b0ff09b06d14a4 /src/backends/neon/test/NeonEndToEndTests.cpp | |
| parent | 1939df5355ebc3994967e9d1cb2c9440a0d3eced (diff) | |
| download | armnn-68a4dbcebbf3c8e759c3f809efbaee2560dd6381.tar.gz | |
IVGCVSW-4053 Enable ArgMinMax EndToEndTest for NEON/CL
* Enabled for Float32 only, as per support in ACL.
Signed-off-by: James Conroy <james.conroy@arm.com>
Change-Id: I251fc832e3058d389ee9bef96856baff89ba6f9a
Diffstat (limited to 'src/backends/neon/test/NeonEndToEndTests.cpp')
| -rw-r--r-- | src/backends/neon/test/NeonEndToEndTests.cpp | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/src/backends/neon/test/NeonEndToEndTests.cpp b/src/backends/neon/test/NeonEndToEndTests.cpp index 26c06758a4..e2b25a14f3 100644 --- a/src/backends/neon/test/NeonEndToEndTests.cpp +++ b/src/backends/neon/test/NeonEndToEndTests.cpp @@ -6,6 +6,7 @@ #include <backendsCommon/test/EndToEndTestImpl.hpp> #include <backendsCommon/test/AbsEndToEndTestImpl.hpp> +#include <backendsCommon/test/ArgMinMaxEndToEndTestImpl.hpp> #include <backendsCommon/test/ComparisonEndToEndTestImpl.hpp> #include <backendsCommon/test/ConcatEndToEndTestImpl.hpp> #include <backendsCommon/test/DepthToSpaceEndToEndTestImpl.hpp> @@ -402,4 +403,55 @@ BOOST_AUTO_TEST_CASE(NeonInstanceNormalizationNchwEndToEndTest2) InstanceNormalizationNchwEndToEndTest2(defaultBackends); } +// ArgMinMax +BOOST_AUTO_TEST_CASE(NeonArgMaxSimpleTest) +{ + ArgMaxEndToEndSimple<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(NeonArgMinSimpleTest) +{ + ArgMinEndToEndSimple<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(NeonArgMaxAxis0Test) +{ + ArgMaxAxis0EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(NeonArgMinAxis0Test) +{ + ArgMinAxis0EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(NeonArgMaxAxis1Test) +{ + ArgMaxAxis1EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(NeonArgMinAxis1Test) +{ + ArgMinAxis1EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(NeonArgMaxAxis2Test) +{ + ArgMaxAxis2EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(NeonArgMinAxis2Test) +{ + ArgMinAxis2EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(NeonArgMaxAxis3Test) +{ + ArgMaxAxis3EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(NeonArgMinAxis3Test) +{ + ArgMinAxis3EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + BOOST_AUTO_TEST_SUITE_END() |