diff options
| author | Kevin Cheng <kevin.cheng@arm.com> | 2021-10-15 20:06:00 +0000 |
|---|---|---|
| committer | Kevin Cheng <kevin.cheng@arm.com> | 2021-10-18 18:50:08 +0000 |
| commit | 5d00c69051bef9f27b60ba136c0efc49a45bf8e6 (patch) | |
| tree | 3b4c6d482e920a5f733409daf2b59ec6d964e513 /reference_model/src/subgraph_traverser.h | |
| parent | cc61be36c3b0f5cd1ea719e129a54fd48a6ee9a2 (diff) | |
| download | reference_model-5d00c69051bef9f27b60ba136c0efc49a45bf8e6.tar.gz | |
Add ERROR_IF to control flow ops.
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: Ifd771171904d1e5a9db3ea1cae3ac9017e971c8c
Diffstat (limited to 'reference_model/src/subgraph_traverser.h')
| -rw-r--r-- | reference_model/src/subgraph_traverser.h | 4 |
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; |