diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-10-31 14:24:02 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-11-01 09:01:49 +0000 |
commit | d1f57736b0dd5168e39725f37188239f8ca024d8 (patch) | |
tree | 98160fa19ba9599cf0315133800a1e931a82e386 /src/backends/reference/test/RefEndToEndTests.cpp | |
parent | 3f4d7104cdd5fd0ca6238afd4d22ac8734fd84ae (diff) | |
download | armnn-d1f57736b0dd5168e39725f37188239f8ca024d8.tar.gz |
IVGCVSW-3698 Add EndToEnd Layer test for ArgMinMax
* Add EndToEnd test implementation for ArgMinMax
* Add EndToEnd tests for Ref
* Fix output data type of ArgMinMax in WorkloadFactory
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I6d07d25bb96ab21422584284046222257ddee43c
Diffstat (limited to 'src/backends/reference/test/RefEndToEndTests.cpp')
-rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 106 |
1 files changed, 106 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index 1968e4da7e..4d8c82d900 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -6,6 +6,7 @@ #include <backendsCommon/test/EndToEndTestImpl.hpp> #include <backendsCommon/test/AbsEndToEndTestImpl.hpp> +#include <backendsCommon/test/ArgMinMaxEndToEndTestImpl.hpp> #include <backendsCommon/test/BatchToSpaceNdEndToEndTestImpl.hpp> #include <backendsCommon/test/ComparisonEndToEndTestImpl.hpp> #include <backendsCommon/test/ConcatEndToEndTestImpl.hpp> @@ -1041,6 +1042,111 @@ BOOST_AUTO_TEST_CASE(RefInstanceNormalizationNchwEndToEndTest2) InstanceNormalizationNchwEndToEndTest2(defaultBackends); } +// ArgMinMax +BOOST_AUTO_TEST_CASE(RefArgMaxSimpleTest) +{ + ArgMaxEndToEndSimple<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMaxSimpleUint8Test) +{ + ArgMaxEndToEndSimple<armnn::DataType::QuantisedAsymm8>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMinSimpleTest) +{ + ArgMinEndToEndSimple<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMinSimpleUint8Test) +{ + ArgMinEndToEndSimple<armnn::DataType::QuantisedAsymm8>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMaxAxis0Test) +{ + ArgMaxAxis0EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMaxAxis0Uint8Test) +{ + ArgMaxAxis0EndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMinAxis0Test) +{ + ArgMinAxis0EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMinAxis0Uint8Test) +{ + + ArgMinAxis0EndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMaxAxis1Test) +{ + ArgMaxAxis1EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMaxAxis1Uint8Test) +{ + ArgMaxAxis1EndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMinAxis1Test) +{ + ArgMinAxis1EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMinAxis1Uint8Test) +{ + + ArgMinAxis1EndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMaxAxis2Test) +{ + ArgMaxAxis2EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMaxAxis2Uint8Test) +{ + ArgMaxAxis2EndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMinAxis2Test) +{ + ArgMinAxis2EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMinAxis2Uint8Test) +{ + + ArgMinAxis2EndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMaxAxis3Test) +{ + ArgMaxAxis3EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMaxAxis3Uint8Test) +{ + ArgMaxAxis3EndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMinAxis3Test) +{ + ArgMinAxis3EndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefArgMinAxis3Uint8Test) +{ + + ArgMinAxis3EndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends); +} + #if !defined(__ANDROID__) // Only run these tests on non Android platforms BOOST_AUTO_TEST_CASE(RefImportNonAlignedPointerTest) |