pub unsafe extern "C" fn mlx_conv_general(
res: *mut mlx_array,
input: mlx_array,
weight: mlx_array,
stride: *const c_int,
stride_num: usize,
padding_lo: *const c_int,
padding_lo_num: usize,
padding_hi: *const c_int,
padding_hi_num: usize,
kernel_dilation: *const c_int,
kernel_dilation_num: usize,
input_dilation: *const c_int,
input_dilation_num: usize,
groups: c_int,
flip: bool,
s: mlx_stream,
) -> c_int