diff options
Diffstat (limited to 'src/backends/reference/test/RefEndToEndTests.cpp')
| -rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index 4f4a161509..330f406265 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -4,6 +4,7 @@ // #include <backendsCommon/test/EndToEndTestImpl.hpp> +#include <backendsCommon/test/GatherEndToEndTestImpl.hpp> #include <backendsCommon/test/MergerTestImpl.hpp> #include <backendsCommon/test/ArithmeticTestImpl.hpp> @@ -416,4 +417,24 @@ BOOST_AUTO_TEST_CASE(RefMergerEndToEndDim3Uint8Test) MergerDim3EndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends); } +BOOST_AUTO_TEST_CASE(RefGatherFloatTest) +{ + GatherEndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefGatherUint8Test) +{ + GatherEndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefGatherMultiDimFloatTest) +{ + GatherMultiDimEndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefGatherMultiDimUint8Test) +{ + GatherMultiDimEndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends); +} + BOOST_AUTO_TEST_SUITE_END() |