diff options
Diffstat (limited to 'src/backends/reference/test/RefLayerSupportTests.cpp')
-rw-r--r-- | src/backends/reference/test/RefLayerSupportTests.cpp | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefLayerSupportTests.cpp b/src/backends/reference/test/RefLayerSupportTests.cpp index b7fbc68d3d..2c7e17da43 100644 --- a/src/backends/reference/test/RefLayerSupportTests.cpp +++ b/src/backends/reference/test/RefLayerSupportTests.cpp @@ -9,6 +9,7 @@ #include <backendsCommon/CpuTensorHandle.hpp> #include <reference/RefWorkloadFactory.hpp> +#include <reference/RefLayerSupport.hpp> #include <backendsCommon/test/LayerTests.hpp> #include <backendsCommon/test/IsLayerSupportedTestImpl.hpp> @@ -32,6 +33,20 @@ BOOST_AUTO_TEST_CASE(IsLayerSupportedLayerTypeMatches) { LayerTypeMatchesTest(); } +BOOST_AUTO_TEST_CASE(IsLayerSupportedReferenceAddition) +{ + armnn::TensorShape shape0 = {1,1,3,4}; + armnn::TensorShape shape1 = {4}; + armnn::TensorShape outShape = {1,1,3,4}; + armnn::TensorInfo in0(shape0, armnn::DataType::Float32); + armnn::TensorInfo in1(shape1, armnn::DataType::Float32); + armnn::TensorInfo out(outShape, armnn::DataType::Float32); + + armnn::RefLayerSupport supportChecker; + std::string reasonNotSupported; + BOOST_CHECK(supportChecker.IsAdditionSupported(in0, in1, out, reasonNotSupported)); +} + BOOST_AUTO_TEST_CASE(IsLayerSupportedFloat16Reference) { |