diff options
Diffstat (limited to 'reference_model/src/subgraph_traverser.h')
-rw-r--r-- | reference_model/src/subgraph_traverser.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/reference_model/src/subgraph_traverser.h b/reference_model/src/subgraph_traverser.h index 4be6c1f..d53a4c0 100644 --- a/reference_model/src/subgraph_traverser.h +++ b/reference_model/src/subgraph_traverser.h @@ -21,6 +21,7 @@ #include "ops/op_factory.h" #include "tensor.h" #include "tosa_serialization_handler.h" +#include <unordered_set> namespace TosaReference { @@ -54,6 +55,7 @@ public: int linkTensorsAndNodes(); int validateGraph(); + int allocateTensor(); int dumpGraph(FILE* out) const; int dumpNextNodeList(FILE* out) const; @@ -99,6 +101,9 @@ private: // lifetime, although the list itself should only contain unique nodes. std::list<GraphNode*> nextNodeList; + // tensor name set which contains all the name used by operator + std::unordered_set<std::string> used_tensor_name_set; + // Maximum number of times to evalute a node before // warning. const int MAX_EVAL_COUNT = 10000; |