Skip to content

Commit ab0e1ef

Browse files
committed
A last one
1 parent a8ea75a commit ab0e1ef

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

mlx/ops.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4206,10 +4206,12 @@ array conv_transpose_general(
42064206
output_padding[i]; // Adjust with output_padding
42074207
}
42084208

4209+
auto ndim = stride.size();
4210+
42094211
return conv_general(
42104212
/* const array& input = */ input,
42114213
/* const array& weight = */ weight,
4212-
/* std::vector<int> stride = */ std::vector(stride.size(), 1),
4214+
/* std::vector<int> stride = */ std::vector(ndim, 1),
42134215
/* std::vector<int> padding_lo = */ std::move(padding_lo),
42144216
/* std::vector<int> padding_hi = */ std::move(padding_hi),
42154217
/* std::vector<int> kernel_dilation = */ std::move(dilation),

0 commit comments

Comments
 (0)