From c5b549b599ff459a29115a48e8f067eaa5891638 Mon Sep 17 00:00:00 2001 From: Michael McGeagh Date: Fri, 7 Aug 2020 11:54:28 +0100 Subject: MLBEDSW-2637 Utilise new tensor and operator funcs add_input_tensor, set_output_tensor, create_const_tensor and create_reshape_tensor have recently been added. This replaces all found existing instances with these new helper functions Signed-off-by: Michael McGeagh Change-Id: If33be8dbf237b2087b562b03cdeb51da1f99a786 --- ethosu/vela/extract_npu_subgraphs.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'ethosu/vela/extract_npu_subgraphs.py') diff --git a/ethosu/vela/extract_npu_subgraphs.py b/ethosu/vela/extract_npu_subgraphs.py index 6747ec98..4adddc17 100644 --- a/ethosu/vela/extract_npu_subgraphs.py +++ b/ethosu/vela/extract_npu_subgraphs.py @@ -80,9 +80,8 @@ def rewrite_tensor_cpu_producer_npu_consumers( op_type = "Const" op = Operation(op_type, orig_tens.name + "_input") op.attrs["npu_block_type"] = NpuBlockType.Default - op.outputs = [new_tens] op.scheduled_pass = startup_init_ps - new_tens.ops = [op] + op.set_output_tensor(new_tens) startup_init_ps.ops.append(op) startup_init_ps.outputs.append(new_tens) -- cgit v1.2.1