aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/nn_graph.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/nn_graph.py')
-rw-r--r--ethosu/vela/nn_graph.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py
index 21cd80b9..58aab611 100644
--- a/ethosu/vela/nn_graph.py
+++ b/ethosu/vela/nn_graph.py
@@ -141,6 +141,8 @@ class Subgraph:
self.placement = placement
self.command_stream_tensor = None
self.flash_tensor = None
+ # Scratch information locally used in the scheduler
+ self.scheduling_info = {}
self.memory_used = {}
self.memory_used_per_type = {}