1 files changed, 4 insertions, 0 deletions
diff --git a/reference_model/src/subgraph_traverser.h b/reference_model/src/subgraph_traverser.h
index d53a4c0..8c66d73 100644
--- a/reference_model/src/subgraph_traverser.h
+++ b/reference_model/src/subgraph_traverser.h
@@ -61,6 +61,10 @@ public:
     int dumpNextNodeList(FILE* out) const;
     int clearAllNodeMarkings();
 
+    std::string getBlockName() const
+    {
+        return block->GetName();
+    }
     int getNumInputTensors() const;
     Tensor* getInputTensor(const unsigned int idx) const;
     Tensor* getInputTensorByName(const std::string name) const;