1 files changed, 10 insertions, 0 deletions
diff --git a/src/armnn/test/TestLayerVisitor.cpp b/src/armnn/test/TestLayerVisitor.cpp
index 932aef6deb..4c028b3709 100644
--- a/src/armnn/test/TestLayerVisitor.cpp
+++ b/src/armnn/test/TestLayerVisitor.cpp
@@ -48,4 +48,14 @@ void TestLayerVisitor::CheckConstTensors(const ConstTensor& expected, const Cons
     }
 }
 
+void TestLayerVisitor::CheckOptionalConstTensors(const Optional<ConstTensor>& expected,
+                                                 const Optional<ConstTensor>& actual)
+{
+    BOOST_CHECK(expected.has_value() == actual.has_value());
+    if (expected.has_value() && actual.has_value())
+    {
+        CheckConstTensors(expected.value(), actual.value());
+    }
+}
+
 } //namespace armnn