diff options
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r-- | ethosu/vela/tensor.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 37fd06ea..2e70d72e 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -632,7 +632,7 @@ class Tensor: self, coord: Optional[Shape] = None, shape4D: Optional[Shape4D] = None ) -> Tuple[Optional[Shape], Optional[Shape]]: if coord is None: - coord = [0] * len(self.storage_shape) + coord = [0] * min(len(self.storage_shape), 4) if shape4D and self.is_standard_fm: augmented_shape = self.get_4D_storage_shape_for_shape(shape4D).as_list() |