diff options
Diffstat (limited to 'ethosu/vela/graph_optimiser.py')
-rw-r--r-- | ethosu/vela/graph_optimiser.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py index a8f68ae1..2bd57ddd 100644 --- a/ethosu/vela/graph_optimiser.py +++ b/ethosu/vela/graph_optimiser.py @@ -996,8 +996,8 @@ def remove_unwanted_reshapes(op, arch): if prev_op_ifm.quantization.is_scaling_equal( prev_op_ofm.quantization ) and cons_op_ifm.quantization.is_scaling_equal(cons_op_ofm.quantization): - op.inputs[0] = prev_op_ifm - op.outputs[0] = cons_op_ofm + op.set_input_tensor(prev_op_ifm, 0) + op.set_output_tensor(cons_op_ofm) return op |