diff options
author | James Ward <james.ward@arm.com> | 2021-10-28 10:01:40 +0100 |
---|---|---|
committer | James Ward <james.ward@arm.com> | 2021-10-28 10:26:42 +0100 |
commit | 3e13434bb787d024e3386ff75f305bfe8cda3288 (patch) | |
tree | eae1d14cdd4a4e083067f0aa3812d959e61865ad /ethosu | |
parent | 0ae2848580d376c66d6a3c3b0fb55b30a0234247 (diff) | |
download | ethos-u-vela-3e13434bb787d024e3386ff75f305bfe8cda3288.tar.gz |
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
Diffstat (limited to 'ethosu')
-rw-r--r-- | ethosu/vela/tflite_graph_optimiser.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tflite_graph_optimiser.py b/ethosu/vela/tflite_graph_optimiser.py index f59eddef..2469a700 100644 --- a/ethosu/vela/tflite_graph_optimiser.py +++ b/ethosu/vela/tflite_graph_optimiser.py @@ -1058,7 +1058,7 @@ def convert_pad(op: Operation, arch, nng): 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 = [] |