commit e1cc3de77668b66ea413c570de7161e2bba89502
author Andreas Nevalainen <andreas.nevalainen@arm.com> Tue Sep 08 15:31:02 2020 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Fri Sep 11 12:18:29 2020 +0000
tree a24d717bc947f64d8f78ddcd84e7852562f79372
parent 2b7a1628ab26b453a0932ce845f41fef8e5ade34

MLBEDSW-2813: Handle None tensor when parsing operators

Signed-off-by: Andreas Nevalainen <andreas.nevalainen@arm.com>
Change-Id: I49c8d283eaa8d44839b94aaf4e90ec35365dc13f

ethosu/vela/nn_graph.py

diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py
index b495828..21cd80b 100644
--- a/ethosu/vela/nn_graph.py
+++ b/ethosu/vela/nn_graph.py
@@ -301,7 +301,7 @@
             all_ops.append(op)
 
         def visit_tensor(tens):
-            if tens in visit_tensor_set:
+            if tens is None or tens in visit_tensor_set:
                 return
             visit_tensor_set.add(tens)
             for op in tens.ops: