Skip to main content

mlx_conv_transpose3d

Function mlx_conv_transpose3d 

Source
pub unsafe extern "C" fn mlx_conv_transpose3d(
    res: *mut mlx_array,
    input: mlx_array,
    weight: mlx_array,
    stride_0: c_int,
    stride_1: c_int,
    stride_2: c_int,
    padding_0: c_int,
    padding_1: c_int,
    padding_2: c_int,
    dilation_0: c_int,
    dilation_1: c_int,
    dilation_2: c_int,
    output_padding_0: c_int,
    output_padding_1: c_int,
    output_padding_2: c_int,
    groups: c_int,
    s: mlx_stream,
) -> c_int