mlx_jvp function

  1. @Native<Int Function(Pointer<mlx_vector_array>, Pointer<mlx_vector_array>, mlx_closure, mlx_vector_array, mlx_vector_array)>(ffi.Pointer<mlx_vector_array>, ffi.Pointer<mlx_vector_array>, mlx_closure, mlx_vector_array, mlx_vector_array)>()
int mlx_jvp(
  1. Pointer<mlx_vector_array> res_0,
  2. Pointer<mlx_vector_array> res_1,
  3. mlx_closure fun,
  4. mlx_vector_array primals,
  5. mlx_vector_array tangents,
)

Implementation

@ffi.Native<
  ffi.Int Function(
    ffi.Pointer<mlx_vector_array>,
    ffi.Pointer<mlx_vector_array>,
    mlx_closure,
    mlx_vector_array,
    mlx_vector_array,
  )
>()
external int mlx_jvp(
  ffi.Pointer<mlx_vector_array> res_0,
  ffi.Pointer<mlx_vector_array> res_1,
  mlx_closure fun,
  mlx_vector_array primals,
  mlx_vector_array tangents,
);