Skip to main content

mlx_conv1d

Function mlx_conv1d 

Source
pub unsafe extern "C" fn mlx_conv1d(
    res: *mut mlx_array,
    input: mlx_array,
    weight: mlx_array,
    stride: c_int,
    padding: c_int,
    dilation: c_int,
    groups: c_int,
    s: mlx_stream,
) -> c_int