diff options
Diffstat (limited to 'ethosu/vela/nn_graph.py')
| -rw-r--r-- | ethosu/vela/nn_graph.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py index 846632df..a43aac2a 100644 --- a/ethosu/vela/nn_graph.py +++ b/ethosu/vela/nn_graph.py @@ -149,7 +149,11 @@ class Subgraph: def __init__(self, name="<unnamed>", placement=PassPlacement.Cpu): self.output_tensors = [] self.input_tensors = [] - self.original_inputs = [] # Preserve the original input order + # Preserve the original input order + self.original_inputs = [] + # Attach virtual outputs to resource variables op + # in order to be able to traverse the graph correctly + self.virtual_outputs = [] self.passes = [] self.cascaded_passes = [] self.name = name |