diff options
author | Derek Lamberti <derek.lamberti@arm.com> | 2019-03-13 14:16:15 +0000 |
---|---|---|
committer | Derek Lamberti <derek.lamberti@arm.com> | 2019-03-20 11:57:44 +0000 |
commit | 50db4e837c885fb645fc4b34373e73054649bceb (patch) | |
tree | e2d4db4294174c99ee85d1e1f47cc460450f4d73 /src/backends/reference/test/RefLayerSupportTests.cpp | |
parent | 03acd6871b2e752f6ea6b1a4e1b739cdae698477 (diff) | |
download | armnn-50db4e837c885fb645fc4b34373e73054649bceb.tar.gz |
IVGCVSW-2814 Extensive ref IsSupported for Activation & Addition
Change-Id: Ib1a795eb129de1ec3f02807a2dff7613d7c6c28d
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
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) { |