diff options
| -rw-r--r-- | ethosu/vela/mark_tensors.py | 2 | ||||
| -rw-r--r-- | ethosu/vela/tensor.py | 2 |
2 files changed, 4 insertions, 0 deletions
diff --git a/ethosu/vela/mark_tensors.py b/ethosu/vela/mark_tensors.py index 1379628a..723bd876 100644 --- a/ethosu/vela/mark_tensors.py +++ b/ethosu/vela/mark_tensors.py @@ -59,6 +59,8 @@ def rewrite_mark_tensor_purpose(op, arch): mark_purpose(tens, arch, TensorPurpose.FeatureMap) weight_tensors = op.get_weight_tensors() for tens in op.inputs: + if tens is None: + continue if tens.purpose != TensorPurpose.Unknown: purpose = tens.purpose elif tens in weight_tensors: 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] |