commit fad720480c3b115664ed5d0dc6028c5020005cf7
author Jacob Bohlin <jacob.bohlin@arm.com> Tue Aug 24 21:51:41 2021 +0200
committer Fredrik Svedberg <fredrik.svedberg@arm.com> Mon Sep 06 09:39:11 2021 +0100
tree 69b9de9be256143bfc2692c4add554b24c47d80d
parent b081d67f3e96896ea0ca7426a18c5fe153419ee0

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 <jacob.bohlin@arm.com>
Change-Id: I2501f35e5668b3085d917751cfc8002d250973d8

ethosu/vela/tensor.py

diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index 8304a65..37fd06e 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -535,7 +535,7 @@
             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