diff options
| author | Jacob Bohlin <jacob.bohlin@arm.com> | 2021-08-24 21:51:41 +0200 |
|---|---|---|
| committer | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2021-09-06 09:39:11 +0100 |
| commit | fad720480c3b115664ed5d0dc6028c5020005cf7 (patch) | |
| tree | 69b9de9be256143bfc2692c4add554b24c47d80d /ethosu/vela/tensor.py | |
| parent | b081d67f3e96896ea0ca7426a18c5fe153419ee0 (diff) | |
| download | ethos-u-vela-fad720480c3b115664ed5d0dc6028c5020005cf7.tar.gz | |
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
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 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 |