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 <michael.mcgeagh@arm.com>
Change-Id: If33be8dbf237b2087b562b03cdeb51da1f99a786
diff --git a/ethosu/vela/tflite_reader.py b/ethosu/vela/tflite_reader.py
index bf3fe95..5e966d1 100644
--- a/ethosu/vela/tflite_reader.py
+++ b/ethosu/vela/tflite_reader.py
@@ -54,9 +54,7 @@
tens.quant_values = tens.quant_values.transpose(reorder)
op = Operation("Const", tens.name)
- op.outputs = [tens]
- tens.ops = [op]
-
+ op.set_output_tensor(tens)
return tens
@@ -81,14 +79,12 @@
TensorError(tens, "This subgraph input tensor has unexpected driving operators.")
op = Operation("Placeholder", tens.name)
- op.outputs = [tens]
- tens.ops = [op]
+ op.set_output_tensor(tens)
for tens in self.tensors:
if not tens.ops:
op = Operation("Const", tens.name)
- op.outputs = [tens]
- tens.ops = [op]
+ op.set_output_tensor(tens)
def get_tensors_from_indices_remove_duplicates(self, indices, warning_str):
tensors = []
@@ -190,8 +186,7 @@
act_op = Operation(activation_function_to_split_out, name + activation_function_to_split_out)
out_tens = op.outputs[0]
intermediate_tens = out_tens.clone("_act_intermediate")
- out_tens.ops = [act_op]
- act_op.outputs = [out_tens]
+ act_op.set_output_tensor(out_tens)
intermediate_tens.ops = [op]
op.outputs[0] = intermediate_tens
act_op.inputs = [intermediate_tens]