aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/test/TestLayerVisitor.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/test/TestLayerVisitor.hpp')
-rw-r--r--src/armnn/test/TestLayerVisitor.hpp10
1 files changed, 9 insertions, 1 deletions
diff --git a/src/armnn/test/TestLayerVisitor.hpp b/src/armnn/test/TestLayerVisitor.hpp
index 16227fb7f5..1998fb9b3b 100644
--- a/src/armnn/test/TestLayerVisitor.hpp
+++ b/src/armnn/test/TestLayerVisitor.hpp
@@ -16,11 +16,19 @@ protected:
void CheckLayerName(const char* name);
+ void CheckLayerPointer(const IConnectableLayer* layer);
+
private:
const char* m_LayerName;
public:
- explicit TestLayerVisitor(const char* name) : m_LayerName(name) {};
+ explicit TestLayerVisitor(const char* name) : m_LayerName(name)
+ {
+ if (name == nullptr)
+ {
+ m_LayerName = "";
+ }
+ };
virtual void VisitInputLayer(const IConnectableLayer* layer,
LayerBindingId id,