commit a21b2e88d19d8cb11a9120d40bacbb594d600565
author Won Jeon <won.jeon@arm.com> Thu Aug 10 10:33:01 2023 +0000
committer Won Jeon <won.jeon@arm.com> Fri Aug 18 15:21:15 2023 -0700
tree 3bc8a40db72a31c1e552a3bd6339627a1175686e
parent e0247481eb1f83f6eb7161d3f7ac2690b180952a

Add DIM operator to reference model

Signed-off-by: Won Jeon <won.jeon@arm.com>
Change-Id: Iea11ee5d3d98773e9c5e9b827593c05afb41ce3b

reference_model/src/tensor.h

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 @@
                         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)
                 {