We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a8ea75a commit ab0e1efCopy full SHA for ab0e1ef
1 file changed
mlx/ops.cpp
@@ -4206,10 +4206,12 @@ array conv_transpose_general(
4206
output_padding[i]; // Adjust with output_padding
4207
}
4208
4209
+ auto ndim = stride.size();
4210
+
4211
return conv_general(
4212
/* const array& input = */ input,
4213
/* const array& weight = */ weight,
- /* std::vector<int> stride = */ std::vector(stride.size(), 1),
4214
+ /* std::vector<int> stride = */ std::vector(ndim, 1),
4215
/* std::vector<int> padding_lo = */ std::move(padding_lo),
4216
/* std::vector<int> padding_hi = */ std::move(padding_hi),
4217
/* std::vector<int> kernel_dilation = */ std::move(dilation),
0 commit comments