diff options
author | Louis Verhaard <louis.verhaard@arm.com> | 2020-09-15 14:05:38 +0200 |
---|---|---|
committer | Louis Verhaard <louis.verhaard@arm.com> | 2020-09-25 08:32:55 +0200 |
commit | 0b9c9a3873da3d368e184308f4f9a4c202e3fb67 (patch) | |
tree | f057bfeacd6c6323cd0e7f33cd7ec33f941b4f8d /ethosu/vela/nn_graph.py | |
parent | 8854dc9088586e1bb0bf2640b2289903dfa3c822 (diff) | |
download | ethos-u-vela-0b9c9a3873da3d368e184308f4f9a4c202e3fb67.tar.gz |
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>
Diffstat (limited to 'ethosu/vela/nn_graph.py')
-rw-r--r-- | ethosu/vela/nn_graph.py | 2 |
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 = {} |