From e1cc3de77668b66ea413c570de7161e2bba89502 Mon Sep 17 00:00:00 2001 From: Andreas Nevalainen Date: Tue, 8 Sep 2020 15:31:02 +0200 Subject: MLBEDSW-2813: Handle None tensor when parsing operators Signed-off-by: Andreas Nevalainen Change-Id: I49c8d283eaa8d44839b94aaf4e90ec35365dc13f --- ethosu/vela/nn_graph.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ethosu/vela') diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py index b495828e..21cd80b9 100644 --- a/ethosu/vela/nn_graph.py +++ b/ethosu/vela/nn_graph.py @@ -301,7 +301,7 @@ class Subgraph: 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: -- cgit v1.2.1