commit 3df5b9624cdf6c2668ea35edc84076f5612bfc15
author Tim Hall <tim.hall@arm.com> Fri Dec 17 14:09:19 2021 +0000
committer tim.hall <tim.hall@arm.com> Fri Dec 17 16:57:50 2021 +0000
tree a73ab69561555890fdf5ee4681d48be90a6a1867
parent 86dcee2667bef8f44912faaf6de47d90ee7dc434

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 <tim.hall@arm.com>
Change-Id: I3ebd7fc4c7ef5c06488a36d8340a17ae6afd4609

ethosu/vela/pass_packing.py

diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py
index 530ad78..1fefdf4 100644
--- a/ethosu/vela/pass_packing.py
+++ b/ethosu/vela/pass_packing.py
@@ -399,6 +399,7 @@
             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)