zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-21 20:46:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Move args in conv_general (#1118)

Browse Source 
Also fix a typo that padding_lo is passed as padding_hi.
This commit is contained in:
Cheng 2024-05-14 21:50:09 +08:00 committed by GitHub
parent 1a7ed5dcb6
commit bf925d9dc7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
python/src/ops.cpp
View File

@ -3353,13 +3353,15 @@ void init_ops(nb::module_& m) {
} }
return conv_general( return conv_general(
/* const array& input = */ input, /* array input = */ std::move(input),
/* const array& weight = */ weight, /* array weight = */ std::move(weight),
/* std::vector<int> stride = */ stride_vec, /* std::vector<int> stride = */ std::move(stride_vec),
/* std::vector<int> padding_lo = */ padding_lo_vec, /* std::vector<int> padding_lo = */ std::move(padding_lo_vec),
/* std::vector<int> padding_hi = */ padding_lo_vec, /* std::vector<int> padding_hi = */ std::move(padding_hi_vec),
/* std::vector<int> kernel_dilation = */ kernel_dilation_vec, /* std::vector<int> kernel_dilation = */
/* std::vector<int> input_dilation = */ input_dilation_vec, std::move(kernel_dilation_vec),
/* std::vector<int> input_dilation = */
std::move(input_dilation_vec),
/* int groups = */ groups, /* int groups = */ groups,
/* bool flip = */ flip, /* bool flip = */ flip,
s); s);