diff options
-rw-r--r-- | ethosu/vela/pass_packing.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py index 530ad78e..1fefdf42 100644 --- a/ethosu/vela/pass_packing.py +++ b/ethosu/vela/pass_packing.py @@ -399,6 +399,7 @@ def pack_into_passes(nng, arch, verbose_packing=False): avgpool_op.ifm_shapes = op.ifm_shapes.copy() avgpool_op.ofm_shapes = op.ofm_shapes.copy() avgpool_op.read_offsets = op.read_offsets.copy() + avgpool_op.read_shapes = op.read_shapes.copy() op.inputs[0] = avgpool_out op_list.insert(0, avgpool_op) |