mlx_conv_general function

  1. @Native<Int Function(Pointer<mlx_array>, mlx_array, mlx_array, Pointer<Int>, Size, Pointer<Int>, Size, Pointer<Int>, Size, Pointer<Int>, Size, Pointer<Int>, Size, Int, Bool, mlx_stream)>(ffi.Pointer<mlx_array>, mlx_array, mlx_array, ffi.Pointer<ffi.Int>, ffi.Size, ffi.Pointer<ffi.Int>, ffi.Size, ffi.Pointer<ffi.Int>, ffi.Size, ffi.Pointer<ffi.Int>, ffi.Size, ffi.Pointer<ffi.Int>, ffi.Size, ffi.Int, ffi.Bool, mlx_stream)>()
int mlx_conv_general(
  1. Pointer<mlx_array> res,
  2. mlx_array input,
  3. mlx_array weight,
  4. Pointer<Int> stride,
  5. int stride_num,
  6. Pointer<Int> padding_lo,
  7. int padding_lo_num,
  8. Pointer<Int> padding_hi,
  9. int padding_hi_num,
  10. Pointer<Int> kernel_dilation,
  11. int kernel_dilation_num,
  12. Pointer<Int> input_dilation,
  13. int input_dilation_num,
  14. int groups,
  15. bool flip,
  16. mlx_stream s,
)

Implementation

@ffi.Native<
  ffi.Int Function(
    ffi.Pointer<mlx_array>,
    mlx_array,
    mlx_array,
    ffi.Pointer<ffi.Int>,
    ffi.Size,
    ffi.Pointer<ffi.Int>,
    ffi.Size,
    ffi.Pointer<ffi.Int>,
    ffi.Size,
    ffi.Pointer<ffi.Int>,
    ffi.Size,
    ffi.Pointer<ffi.Int>,
    ffi.Size,
    ffi.Int,
    ffi.Bool,
    mlx_stream,
  )
>()
external int mlx_conv_general(
  ffi.Pointer<mlx_array> res,
  mlx_array input,
  mlx_array weight,
  ffi.Pointer<ffi.Int> stride,
  int stride_num,
  ffi.Pointer<ffi.Int> padding_lo,
  int padding_lo_num,
  ffi.Pointer<ffi.Int> padding_hi,
  int padding_hi_num,
  ffi.Pointer<ffi.Int> kernel_dilation,
  int kernel_dilation_num,
  ffi.Pointer<ffi.Int> input_dilation,
  int input_dilation_num,
  int groups,
  bool flip,
  mlx_stream s,
);