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

add dilation for conv 3d layers + test for 3d conv w/ dilation (#1802)

Browse Source
This commit is contained in:
Awni Hannun
2025-01-28 06:17:07 -08:00
committed by GitHub
parent ccb61d7aae
commit 1017ac4a9e
4 changed files with 22 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/ops.cpp
View File

@@ -3540,7 +3540,7 @@ Shape conv_out_shape(
if (out_shape[i] <= 0) {
std::ostringstream msg;
msg << "[conv] Spatial dimensions of input after padding "
msg << "[conv] Spatial dimensions of input after padding"
<< " cannot be smaller than weight spatial dimensions."
<< " Got error at axis " << i << " for input with shape " << in_shape
<< ", padding low " << pads_lo << ", padding high " << pads_hi