mlx_custom_function function

  1. @Native<Int Function(Pointer<mlx_closure>, mlx_closure, mlx_closure_custom, mlx_closure_custom_jvp, mlx_closure_custom_vmap)>(ffi.Pointer<mlx_closure>, mlx_closure, mlx_closure_custom, mlx_closure_custom_jvp, mlx_closure_custom_vmap)>()
int mlx_custom_function(
  1. Pointer<mlx_closure> res,
  2. mlx_closure fun,
  3. mlx_closure_custom fun_vjp,
  4. mlx_closure_custom_jvp fun_jvp,
  5. mlx_closure_custom_vmap fun_vmap,
)

Implementation

@ffi.Native<
  ffi.Int Function(
    ffi.Pointer<mlx_closure>,
    mlx_closure,
    mlx_closure_custom,
    mlx_closure_custom_jvp,
    mlx_closure_custom_vmap,
  )
>()
external int mlx_custom_function(
  ffi.Pointer<mlx_closure> res,
  mlx_closure fun,
  mlx_closure_custom fun_vjp,
  mlx_closure_custom_jvp fun_jvp,
  mlx_closure_custom_vmap fun_vmap,
);