diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-01-29 09:56:02 +0000 |
---|---|---|
committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-01-29 09:57:31 +0000 |
commit | c6a41ffa25d468a69465e1a7b22b280b029f65a2 (patch) | |
tree | fe879de0b7d7bdea208600f8a7122f3988bcf1df /src/backends/cl/test/ClEndToEndTests.cpp | |
parent | eed85925f893256f8e813423de0f3c5fc79179ae (diff) | |
download | armnn-c6a41ffa25d468a69465e1a7b22b280b029f65a2.tar.gz |
IVGCVSW-2443 Add Cl support for Greater operator
Change-Id: I7eb3b2e712dbaa5c13f6cf3a1da63df4d2bd4de5
Diffstat (limited to 'src/backends/cl/test/ClEndToEndTests.cpp')
-rw-r--r-- | src/backends/cl/test/ClEndToEndTests.cpp | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/src/backends/cl/test/ClEndToEndTests.cpp b/src/backends/cl/test/ClEndToEndTests.cpp index ba94353049..bc0bad2fa6 100644 --- a/src/backends/cl/test/ClEndToEndTests.cpp +++ b/src/backends/cl/test/ClEndToEndTests.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> @@ -47,4 +48,45 @@ BOOST_AUTO_TEST_CASE(ClMergerEndToEndDim3Uint8Test) MergerDim3EndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends); } +BOOST_AUTO_TEST_CASE(ClGreaterSimpleEndToEndTest) +{ + 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(ClGreaterSimpleEndToEndUint8Test) +{ + 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(ClGreaterBroadcastEndToEndTest) +{ + 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(ClGreaterBroadcastEndToEndUint8Test) +{ + 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_SUITE_END() |