diff options
author | erik.andersson@arm.com <erik.andersson@arm.com> | 2021-03-24 10:32:09 +0100 |
---|---|---|
committer | erik.andersson@arm.com <erik.andersson@arm.com> | 2021-03-30 09:20:00 +0200 |
commit | 3438c929528583bc019055ad7057c08271b0cee7 (patch) | |
tree | ea0f8af0e5f23e567c6d31f4f7c639a0e94f6dbf /ethosu/vela/nn_graph.py | |
parent | 9bb1e2ed361286769c362f002910d1dbd1736b05 (diff) | |
download | ethos-u-vela-3438c929528583bc019055ad7057c08271b0cee7.tar.gz |
MLBEDSW-4219: Add tensor allocation info to summary
Added the theoretically minimum max memory usage and
the allocator overhead to the Vela summary.
Signed-off-by: erik.andersson@arm.com <erik.andersson@arm.com>
Change-Id: If373dfeaac50d6f8b56554d435bf22af2c3acda3
Diffstat (limited to 'ethosu/vela/nn_graph.py')
-rw-r--r-- | ethosu/vela/nn_graph.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py index c45d0e3e..19734873 100644 --- a/ethosu/vela/nn_graph.py +++ b/ethosu/vela/nn_graph.py @@ -154,6 +154,7 @@ class Subgraph: self.memory_used = {} self.memory_used_per_type = {} + self.min_mem_usage = 0 def __str__(self): return "<nng.Subgraph '%s', n_passes=%d, n_cascaded_passes=%d>" % ( |