diff options
author | James Ward <james.ward@arm.com> | 2021-10-14 12:58:02 +0100 |
---|---|---|
committer | James Ward <james.ward@arm.com> | 2021-11-11 13:04:08 +0000 |
commit | 9338978eae5656005d8ac909b41aa5abac7a83a7 (patch) | |
tree | 70fe15e282630cb9528281cf36170e70d07cd52f /ethosu/vela/extract_npu_subgraphs.py | |
parent | cc8569fcd243d7c96a20e0f531f6f97a90df83f7 (diff) | |
download | ethos-u-vela-9338978eae5656005d8ac909b41aa5abac7a83a7.tar.gz |
MLBEDSW-5410 Clarify "Subgraph IO Summary" output
* Add small aesthetic changes to summary
* Move "_cpu" suffix from cloned tensor to original tensor such that suffix is no longer externally visible
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I97427561bd9acb04765ae9de6278760511278118
Diffstat (limited to 'ethosu/vela/extract_npu_subgraphs.py')
-rw-r--r-- | ethosu/vela/extract_npu_subgraphs.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ethosu/vela/extract_npu_subgraphs.py b/ethosu/vela/extract_npu_subgraphs.py index b536cfba..d802b513 100644 --- a/ethosu/vela/extract_npu_subgraphs.py +++ b/ethosu/vela/extract_npu_subgraphs.py @@ -104,7 +104,8 @@ def rewrite_tensor_npu_producer_cpu_consumers( orig_tens, call_ps, startup_init_ps, npu_subgraph, cpu_subgraph, subgraph_for_pass ): - new_tens = orig_tens.clone("_cpu") + new_tens = orig_tens.clone("") + orig_tens.name = orig_tens.name + "_cpu" npu_subgraph.output_tensors.append(orig_tens) call_ps.outputs.append(new_tens) |