diff options
Diffstat (limited to 'ethosu/vela/mark_tensors.py')
-rw-r--r-- | ethosu/vela/mark_tensors.py | 2 |
1 files changed, 2 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: |