diff options
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) |