diff options
Diffstat (limited to 'reference_model/src/tensor.h')
-rw-r--r-- | reference_model/src/tensor.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/reference_model/src/tensor.h b/reference_model/src/tensor.h index f59a5e1..74f57ed 100644 --- a/reference_model/src/tensor.h +++ b/reference_model/src/tensor.h @@ -811,6 +811,13 @@ public: return new Tensor6<int64_t>(tensorName_, dtype_, shape_); } break; + case TOSA_REF_TYPE_SHAPE: + // if shape information is not already set, set it here. + if (shape_.size() == 0) + { + shape_ = { 1 }; + } + return new Tensor0<int64_t>(tensorName_, dtype_, shape_); case TOSA_REF_TYPE_BOOL: switch (rank) { |