mlx_conv_transpose3d function

  1. @Native<Int Function(Pointer<mlx_array>, mlx_array, mlx_array, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, mlx_stream)>(ffi.Pointer<mlx_array>, mlx_array, mlx_array, ffi.Int, ffi.Int, ffi.Int, ffi.Int, ffi.Int, ffi.Int, ffi.Int, ffi.Int, ffi.Int, ffi.Int, ffi.Int, ffi.Int, ffi.Int, mlx_stream)>()
int mlx_conv_transpose3d(
  1. Pointer<mlx_array> res,
  2. mlx_array input,
  3. mlx_array weight,
  4. int stride_0,
  5. int stride_1,
  6. int stride_2,
  7. int padding_0,
  8. int padding_1,
  9. int padding_2,
  10. int dilation_0,
  11. int dilation_1,
  12. int dilation_2,
  13. int output_padding_0,
  14. int output_padding_1,
  15. int output_padding_2,
  16. int groups,
  17. mlx_stream s,
)

Implementation

@ffi.Native<
  ffi.Int Function(
    ffi.Pointer<mlx_array>,
    mlx_array,
    mlx_array,
    ffi.Int,
    ffi.Int,
    ffi.Int,
    ffi.Int,
    ffi.Int,
    ffi.Int,
    ffi.Int,
    ffi.Int,
    ffi.Int,
    ffi.Int,
    ffi.Int,
    ffi.Int,
    ffi.Int,
    mlx_stream,
  )
>()
external int mlx_conv_transpose3d(
  ffi.Pointer<mlx_array> res,
  mlx_array input,
  mlx_array weight,
  int stride_0,
  int stride_1,
  int stride_2,
  int padding_0,
  int padding_1,
  int padding_2,
  int dilation_0,
  int dilation_1,
  int dilation_2,
  int output_padding_0,
  int output_padding_1,
  int output_padding_2,
  int groups,
  mlx_stream s,
);