diff options
Diffstat (limited to 'ethosu/vela/tosa_graph_optimiser.py')
-rw-r--r-- | ethosu/vela/tosa_graph_optimiser.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ethosu/vela/tosa_graph_optimiser.py b/ethosu/vela/tosa_graph_optimiser.py index bcb4aac8..19244c27 100644 --- a/ethosu/vela/tosa_graph_optimiser.py +++ b/ethosu/vela/tosa_graph_optimiser.py @@ -357,6 +357,7 @@ def rewrite_activation(op, arch, nng): return op + def rewrite_rescale(op, arch, nng): if op.type == Op.Rescale: ifm = op.ifm @@ -364,7 +365,6 @@ def rewrite_rescale(op, arch, nng): # some error checking assert len(ifm.ops) == 1 - prev_op = ifm.ops[0] input_zp = op.attrs["input_zp"] output_zp = op.attrs["output_zp"] @@ -409,6 +409,7 @@ def rewrite_rescale(op, arch, nng): return op + def convert_pad_in_width(op): """ Rewrites PAD operator to an add that copies the IFM to the OFM |