diff options
| author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-05-27 09:15:11 +0200 |
|---|---|---|
| committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-06-25 11:42:56 +0200 |
| commit | eca2e95e1fea150d8a942f8b5f0a4d9d7aefebc1 (patch) | |
| tree | 438b385f1ded3c18c3b84d2204a57c39be6be34a /ethosu/vela/nn_graph.py | |
| parent | eec4e50e19cb5522640eae5fd4566917dc2a7b9d (diff) | |
| download | ethos-u-vela-eca2e95e1fea150d8a942f8b5f0a4d9d7aefebc1.tar.gz | |
MLBEDSW-2306 Added more supported mem-cfgs
Additional supported memory configurations:
-Permanent_storage = DRAM
-Tensor arena either in DRAM or SRAM
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I20beb7151e306bfdba540e7c0b2a7b478b4d94e1
Diffstat (limited to 'ethosu/vela/nn_graph.py')
| -rw-r--r-- | ethosu/vela/nn_graph.py | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py index ea35c087..247e6cce 100644 --- a/ethosu/vela/nn_graph.py +++ b/ethosu/vela/nn_graph.py @@ -137,6 +137,7 @@ class Subgraph: self.flash_tensor = None self.memory_used = {} + self.memory_used_per_type = {} def __str__(self): return "<nng.Subgraph '%s', n_passes=%d, n_cascaded_passes=%d>" % ( @@ -349,9 +350,15 @@ class Subgraph: for idx, op in enumerate(all_ops): print(idx, op.type, op.name) for idx, tens in enumerate(op.inputs): - print(" Input %02d %20s %20s %s" % (idx, tens.purpose.name, tens.mem_area.name, tens)) + print( + " Input %02d %20s %20s %20s %s" + % (idx, tens.purpose.name, tens.mem_area.name, tens.mem_type.name, tens) + ) for idx, tens in enumerate(op.outputs): - print(" Output %02d %20s %20s %s" % (idx, tens.purpose.name, tens.mem_area.name, tens)) + print( + " Output %02d %20s %20s %20s %s" + % (idx, tens.purpose.name, tens.mem_area.name, tens.mem_type.name, tens) + ) print() def print_graph_with_tensor_quantization(self): |