diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-09-29 14:09:43 +0200 |
---|---|---|
committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-09-29 14:56:58 +0200 |
commit | 7ad862ada22dc00c0306ba0f1eafbef98f23c77c (patch) | |
tree | a4da05ea0d0f2bd6208374b262ffc55ca7b1fe78 | |
parent | f03bad31c194d1a24ce808223f1b00310a7325e2 (diff) | |
download | ethos-u-vela-7ad862ada22dc00c0306ba0f1eafbef98f23c77c.tar.gz |
MLBEDSW-3032 Fixed issue in removal of reshapes
Fixed issue in removal of reshapes
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: Id6081de8d6b7b6815cc5e56881c20e075214c407
-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 |