diff options
Diffstat (limited to 'src/armnn/test/TestLayerVisitor.cpp')
-rw-r--r-- | src/armnn/test/TestLayerVisitor.cpp | 10 |
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 |