mlx_custom_vjp function

  1. @Native<Int Function(Pointer<mlx_closure>, mlx_closure, mlx_closure_custom)>(ffi.Pointer<mlx_closure>, mlx_closure, mlx_closure_custom)>()
int mlx_custom_vjp(
  1. Pointer<mlx_closure> res,
  2. mlx_closure fun,
  3. mlx_closure_custom fun_vjp
)

Implementation

@ffi.Native<
  ffi.Int Function(ffi.Pointer<mlx_closure>, mlx_closure, mlx_closure_custom)
>()
external int mlx_custom_vjp(
  ffi.Pointer<mlx_closure> res,
  mlx_closure fun,
  mlx_closure_custom fun_vjp,
);