diff options
-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 = [] |