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>
diff --git a/ethosu/vela/mark_tensors.py b/ethosu/vela/mark_tensors.py
index 1379628..723bd87 100644
--- a/ethosu/vela/mark_tensors.py
+++ b/ethosu/vela/mark_tensors.py
@@ -59,6 +59,8 @@
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: