diff options
Diffstat (limited to 'reference_model/src/subgraph_traverser.h')
-rw-r--r-- | reference_model/src/subgraph_traverser.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/reference_model/src/subgraph_traverser.h b/reference_model/src/subgraph_traverser.h index 00989ee..ef6ea42 100644 --- a/reference_model/src/subgraph_traverser.h +++ b/reference_model/src/subgraph_traverser.h @@ -49,7 +49,8 @@ public: int linkTensorsAndNodes(); int validateGraph(); - int allocateTensor(); + int allocateInputTensors(); + int allocateTensor(std::string name); int dumpGraph(FILE* out) const; int dumpNextNodeList(FILE* out) const; |