From 7ad862ada22dc00c0306ba0f1eafbef98f23c77c Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Tue, 29 Sep 2020 14:09:43 +0200 Subject: MLBEDSW-3032 Fixed issue in removal of reshapes Fixed issue in removal of reshapes Signed-off-by: Patrik Gustavsson Change-Id: Id6081de8d6b7b6815cc5e56881c20e075214c407 --- ethosu/vela/graph_optimiser.py | 4 ++-- 1 file 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 -- cgit v1.2.1