diff options
author | kevmay01 <kevin.may@arm.com> | 2019-01-28 08:37:25 +0000 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2019-01-29 08:28:32 +0000 |
commit | eed85925f893256f8e813423de0f3c5fc79179ae (patch) | |
tree | 69d7421e542c13266e17985165760b9d831095be /src/backends/neon/test/NeonEndToEndTests.cpp | |
parent | cd7544e4ea0dc9fdee80447688bf4b8d64cc2ec8 (diff) | |
download | armnn-eed85925f893256f8e813423de0f3c5fc79179ae.tar.gz |
IVGCVSW-2447 Add Greater Workload to Neon
Change-Id: Ic165061cc61534e7e391a398ea7c2918240e2d3b
Diffstat (limited to 'src/backends/neon/test/NeonEndToEndTests.cpp')
-rw-r--r-- | src/backends/neon/test/NeonEndToEndTests.cpp | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/src/backends/neon/test/NeonEndToEndTests.cpp b/src/backends/neon/test/NeonEndToEndTests.cpp index 665791a36a..deaa8eb81b 100644 --- a/src/backends/neon/test/NeonEndToEndTests.cpp +++ b/src/backends/neon/test/NeonEndToEndTests.cpp @@ -5,6 +5,7 @@ #include <backendsCommon/test/EndToEndTestImpl.hpp> #include <backendsCommon/test/MergerTestImpl.hpp> +#include <backendsCommon/test/ArithmeticTestImpl.hpp> #include <boost/test/unit_test.hpp> @@ -51,6 +52,46 @@ BOOST_AUTO_TEST_CASE(FallbackToCpuRef) BOOST_TEST(runtime->LoadNetwork(netId, std::move(optNet)) == Status::Success); } +BOOST_AUTO_TEST_CASE(NeonGreaterSimpleEndToEndTest) +{ + const std::vector<uint8_t> expectedOutput({ 0, 0, 0, 0, 1, 1, 1, 1, + 0, 0, 0, 0, 0, 0, 0, 0 }); + + ArithmeticSimpleEndToEnd<armnn::DataType::Float32, armnn::DataType::Boolean>(defaultBackends, + LayerType::Greater, + expectedOutput); +} + +BOOST_AUTO_TEST_CASE(NeonGreaterSimpleEndToEndUint8Test) +{ + const std::vector<uint8_t> expectedOutput({ 0, 0, 0, 0, 1, 1, 1, 1, + 0, 0, 0, 0, 0, 0, 0, 0 }); + + ArithmeticSimpleEndToEnd<armnn::DataType::QuantisedAsymm8, armnn::DataType::Boolean>(defaultBackends, + LayerType::Greater, + expectedOutput); +} + +BOOST_AUTO_TEST_CASE(NeonGreaterBroadcastEndToEndTest) +{ + const std::vector<uint8_t> expectedOutput({ 0, 1, 0, 0, 0, 1, + 1, 1, 1, 1, 1, 1 }); + + ArithmeticBroadcastEndToEnd<armnn::DataType::Float32, armnn::DataType::Boolean>(defaultBackends, + LayerType::Greater, + expectedOutput); +} + +BOOST_AUTO_TEST_CASE(NeonGreaterBroadcastEndToEndUint8Test) +{ + const std::vector<uint8_t> expectedOutput({ 0, 1, 0, 0, 0, 1, + 1, 1, 1, 1, 1, 1 }); + + ArithmeticBroadcastEndToEnd<armnn::DataType::QuantisedAsymm8, armnn::DataType::Boolean>(defaultBackends, + LayerType::Greater, + expectedOutput); +} + BOOST_AUTO_TEST_CASE(NeonMergerEndToEndDim0Test) { MergerDim0EndToEnd<armnn::DataType::Float32>(defaultBackends); |