diff options
-rw-r--r-- | ethosu/vela/nn_graph.py | 2 |
1 files changed, 1 insertions, 1 deletions
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: |