From 50db4e837c885fb645fc4b34373e73054649bceb Mon Sep 17 00:00:00 2001 From: Derek Lamberti Date: Wed, 13 Mar 2019 14:16:15 +0000 Subject: IVGCVSW-2814 Extensive ref IsSupported for Activation & Addition Change-Id: Ib1a795eb129de1ec3f02807a2dff7613d7c6c28d Signed-off-by: Derek Lamberti --- src/backends/reference/test/RefLayerSupportTests.cpp | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'src/backends/reference/test') 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 #include +#include #include #include @@ -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) { -- cgit v1.2.1