From fad720480c3b115664ed5d0dc6028c5020005cf7 Mon Sep 17 00:00:00 2001 From: Jacob Bohlin Date: Tue, 24 Aug 2021 21:51:41 +0200 Subject: MLBEDSW-4975 Fix semodepth asserts This commit fixes one assert regarding rolling buffers for 3D tensors. It also addresses another issue where the incorrect weight buffering was proposed for cascaded operators. Signed-off-by: Jacob Bohlin Change-Id: I2501f35e5668b3085d917751cfc8002d250973d8 --- ethosu/vela/tensor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ethosu/vela/tensor.py') diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 8304a657..37fd06ea 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -535,7 +535,7 @@ class Tensor: assert param_a is not None shp[-1] = min(shp[-1], param_a * 2) else: - shp = list(self.storage_shape) + shp = full_shape(4, self.storage_shape, 1) if sub_purpose == TensorSubPurpose.RollingBufferX: assert len(shp) == 4 assert param_a is not None -- cgit v1.2.1