diff options
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r-- | ethosu/vela/tensor.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 45518b4e..0d299e15 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -446,6 +446,8 @@ class Tensor: except TypeError: pass + if shape_len > 4: + return self.storage_rounding_quantum = arch.storage_rounding_quantums[self.format] self.storage_rounding_quantum = self.storage_rounding_quantum[-shape_len:] self.brick_size = arch.brick_sizes[self.format] |