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/tensor.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/tensor.h')
| -rw-r--r-- | reference_model/src/tensor.h | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/reference_model/src/tensor.h b/reference_model/src/tensor.h index b68a9b6..21cf148 100644 --- a/reference_model/src/tensor.h +++ b/reference_model/src/tensor.h @@ -96,6 +96,12 @@ public: return shape; } + void setDimSize(size_t dim, uint32_t new_size) + { + this->shape[dim] = new_size; + return; + } + std::string getShapeAsString() const { std::string shape_str("["); @@ -269,7 +275,7 @@ public: protected: const std::string tensorName; const DType serializationDtype; - const std::vector<int> shape; + std::vector<int> shape; const TOSA_REF_TYPE tensorDtype; int isValid; int isSubgraphInput; |