commit 50db4e837c885fb645fc4b34373e73054649bceb
author Derek Lamberti <derek.lamberti@arm.com> Wed Mar 13 14:16:15 2019 +0000
committer Derek Lamberti <derek.lamberti@arm.com> Wed Mar 20 11:57:44 2019 +0000
tree e2d4db4294174c99ee85d1e1f47cc460450f4d73
parent 03acd6871b2e752f6ea6b1a4e1b739cdae698477

IVGCVSW-2814 Extensive ref IsSupported for Activation & Addition

Change-Id: Ib1a795eb129de1ec3f02807a2dff7613d7c6c28d
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>

src/backends/reference/test/RefLayerSupportTests.cpp

diff --git a/src/backends/reference/test/RefLayerSupportTests.cpp b/src/backends/reference/test/RefLayerSupportTests.cpp
index b7fbc68..2c7e17da 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 @@
 {
     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)
 {