diff options
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r-- | ethosu/vela/tensor.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 677757ca..8304a657 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -478,6 +478,7 @@ class Tensor: if shape_len > 4: return + assert not (self.needs_linear_format and fmt == TensorFormat.NHCWB16) self.storage_rounding_quantum = arch.storage_rounding_quantums[self.format] self.storage_rounding_quantum = tuple(self.storage_rounding_quantum[-shape_len:]) self.brick_size = arch.brick_sizes[self.format] |