diff options
| author | Jerry Ge <jerry.ge@arm.com> | 2023-04-21 22:49:57 +0000 |
|---|---|---|
| committer | Jerry Ge <jerry.ge@arm.com> | 2023-05-17 22:46:57 +0000 |
| commit | 264f7faa59709ffa8117541f5d55c99c5dba967d (patch) | |
| tree | ae767b3e4375ab87d4323f18b63239a84ac857db /reference_model/src/subgraph_traverser.h | |
| parent | 7e5968166a5105da30bc11c9241f271cb3dc1da9 (diff) | |
| download | reference_model-264f7faa59709ffa8117541f5d55c99c5dba967d.tar.gz | |
Add support for one dimension of size -1 in ReshapeOp
Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I0ef7607f4266296a1204c5cccdb5be36f345b5ba
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 543b008..1cf582e 100644 --- a/reference_model/src/subgraph_traverser.h +++ b/reference_model/src/subgraph_traverser.h @@ -63,6 +63,10 @@ public: { return block->GetRegionName(); } + TosaSerializationHandler* getTsh() const + { + return tsh; + } int getNumInputTensors() const; Tensor* getInputTensor(const unsigned int idx) const; Tensor* getInputTensorByName(const std::string name) const; |