diff options
| author | Louis Verhaard <louis.verhaard@arm.com> | 2020-11-16 16:37:11 +0100 |
|---|---|---|
| committer | Louis Verhaard <louis.verhaard@arm.com> | 2020-11-17 09:59:14 +0100 |
| commit | 0411edb8947bc2993b1f9498ef43d5b40a2a6305 (patch) | |
| tree | de323b4c1c1abffd2d4f7b8da0547df0735d2b73 /ethosu/vela/tensor.py | |
| parent | 15a8e803844b286fe9533e1cf703c76a77b090a8 (diff) | |
| download | ethos-u-vela-0411edb8947bc2993b1f9498ef43d5b40a2a6305.tar.gz | |
MLBEDSW-3493: bug fixes in mark_tensors
None inputs and unsupported tensor shapes caused asserts when
marking tensor purpose/format.
Change-Id: I4498b61576f529c1a594341cfbb6ba278c6e7ec5
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
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] |