From 3df5b9624cdf6c2668ea35edc84076f5612bfc15 Mon Sep 17 00:00:00 2001 From: Tim Hall Date: Fri, 17 Dec 2021 14:09:19 +0000 Subject: MLBEDSW-5834: split shape is None when split offset is not - This bug causes an exception to occur when trying to index split shape in Box.transform_with_strides_and_skirt() - The bug was due to the read shapes not being initialised when creating a primary op in pass packing Signed-off-by: Tim Hall Change-Id: I3ebd7fc4c7ef5c06488a36d8340a17ae6afd4609 --- ethosu/vela/pass_packing.py | 1 + 1 file changed, 1 insertion(+) 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) -- cgit v1.2.1