aboutsummaryrefslogtreecommitdiff
path: root/pseudocode/operators/DIM.tosac
diff options
context:
space:
mode:
Diffstat (limited to 'pseudocode/operators/DIM.tosac')
-rw-r--r--pseudocode/operators/DIM.tosac2
1 files changed, 1 insertions, 1 deletions
diff --git a/pseudocode/operators/DIM.tosac b/pseudocode/operators/DIM.tosac
index 7131173..2105c62 100644
--- a/pseudocode/operators/DIM.tosac
+++ b/pseudocode/operators/DIM.tosac
@@ -8,4 +8,4 @@
// by a licensing agreement from ARM Limited.
ERROR_IF(axis >= rank(shape));
-tensor_write<shape_t>(output, [], [], shape_dim(shape, axis));
+output[0] = shape_dim(shape, axis);