diff options
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() |