Skip to content

Commit 8c47c70

Browse files
author
Alejandro Gaston Alvarez Franceschi
committed
Fix padding
1 parent 8dc23fe commit 8c47c70

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

Diff for: coremltools/converters/mil/mil/passes/defs/lower_complex_dialect_ops.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -482,8 +482,9 @@ def _istft(
482482
# We need to adapt last dimension
483483
if length is not None:
484484
if length > expected_output_signal_len:
485-
real_result = mb.pad(x=real_result, pad=, mode="constant", constant_val=0, before_op=before_op)
486-
imag_result = mb.pad(x=imag_result, pad=, mode="constant", constant_val=0, before_op=before_op)
485+
right_pad = mb.fill(shape=(channels, expected_output_signal_len - length), value=0., before_op=before_op)
486+
real_result = mb.stack(x=(real_result, right_pad), axis=1, before_op=before_op)
487+
imag_result = mb.stack(x=(imag_result, right_pad), axis=1, before_op=before_op)
487488
elif length < expected_output_signal_len:
488489
real_result = mb.slice_by_size(x=real_result, begin=[0], size=[length], before_op=before_op)
489490
imag_result = mb.slice_by_size(x=imag_result, begin=[0], size=[length], before_op=before_op)

0 commit comments

Comments
 (0)