Skip to main content

mlx_conv_general

Function mlx_conv_general 

Source
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