MLBEDSW-2337: Intermediate feature maps in fast storage
Attempts to use fast storage for feature maps used in between
cascaded passes.
This is only relevant for system configurations where feature maps
are by default not placed in SRAM, but there is SRAM for fast storage.
Change-Id: I207b7cf32cfcb5bea3e6b93c2da1161c4af5221d
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py
index 21cd80b..58aab61 100644
--- a/ethosu/vela/nn_graph.py
+++ b/ethosu/vela/nn_graph.py
@@ -141,6 +141,8 @@
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 = {}