MLBEDSW-5440 MLCE: Vela crashing due to mismatch in padding shape in OFM
Change convert_pad optimiser to use op.ifm_shapes attribute in place of
the fickle op.ifm.shape (which in this case had changed due to the
optimised-out reshape)
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I13fbd846ac8d3342afd7844d1041cfa15aaae124
diff --git a/ethosu/vela/tflite_graph_optimiser.py b/ethosu/vela/tflite_graph_optimiser.py
index f59edde..2469a70 100644
--- a/ethosu/vela/tflite_graph_optimiser.py
+++ b/ethosu/vela/tflite_graph_optimiser.py
@@ -1058,7 +1058,7 @@
ifm = op.ifm
assert ifm is not None
- ifm_shape = Shape4D(ifm.shape)
+ ifm_shape = op.ifm_shapes[0]
ofm = op.ofm
assert ofm is not None
ofm.ops = []