diff options
author | Tai Ly <tai.ly@arm.com> | 2023-09-07 20:49:09 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-09-15 18:10:01 +0000 |
commit | cf84bc9cccbd5dc2fceae1a81c579e41be3c9a06 (patch) | |
tree | aff6bab02c36c095a62381ac8f68d185bdccbe73 /reference_model/src/subgraph_traverser.h | |
parent | 00f55bf46fe36bebe44e1365becbeb1e0d9e90c9 (diff) | |
download | reference_model-cf84bc9cccbd5dc2fceae1a81c579e41be3c9a06.tar.gz |
[reference_model] Support StatefulOps and the tests for CallOnceOp
Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: I03cb878736ccd7e1f5e1f780d7171949a19a9de2
Diffstat (limited to 'reference_model/src/subgraph_traverser.h')
-rw-r--r-- | reference_model/src/subgraph_traverser.h | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/reference_model/src/subgraph_traverser.h b/reference_model/src/subgraph_traverser.h index ef6ea42..d6b0e8d 100644 --- a/reference_model/src/subgraph_traverser.h +++ b/reference_model/src/subgraph_traverser.h @@ -74,10 +74,14 @@ public: int getNumOutputTensors() const; Tensor* getOutputTensor(const unsigned int idx) const; Tensor* getOutputTensorByName(const std::string name) const; + int getNumVariableTensors() const; + Tensor* getVariableTensor(const unsigned int idx) const; + Tensor* getVariableTensorByName(const std::string name) const; + int registerVariableTensor(Tensor* tensor); int addToNextNodeList(GraphNode*); private: - int addTensor(Tensor* ct); + int addTensor(const TosaSerializationTensor* ts); int addNode(GraphNode* cn); Tensor* findTensorByName(const std::string& name) const; @@ -103,6 +107,9 @@ private: // The subset of tensors that are also output tensors std::vector<Tensor*> outputTensors; + // The subset of tensors that are also variable tensors + std::vector<Tensor*> variableTensors; + // The definitive list of all nodes in the graph std::vector<GraphNode*> nodes; |