mlx_detail_vmap_replace function

  1. @Native<Int Function(Pointer<mlx_vector_array>, mlx_vector_array, mlx_vector_array, mlx_vector_array, Pointer<Int>, Size, Pointer<Int>, Size)>(ffi.Pointer<mlx_vector_array>, mlx_vector_array, mlx_vector_array, mlx_vector_array, ffi.Pointer<ffi.Int>, ffi.Size, ffi.Pointer<ffi.Int>, ffi.Size)>()
int mlx_detail_vmap_replace(
  1. Pointer<mlx_vector_array> res,
  2. mlx_vector_array inputs,
  3. mlx_vector_array s_inputs,
  4. mlx_vector_array s_outputs,
  5. Pointer<Int> in_axes,
  6. int in_axes_num,
  7. Pointer<Int> out_axes,
  8. int out_axes_num,
)

\defgroup transforms_impl Implementation detail operations / /**@{

Implementation

@ffi.Native<
  ffi.Int Function(
    ffi.Pointer<mlx_vector_array>,
    mlx_vector_array,
    mlx_vector_array,
    mlx_vector_array,
    ffi.Pointer<ffi.Int>,
    ffi.Size,
    ffi.Pointer<ffi.Int>,
    ffi.Size,
  )
>()
external int mlx_detail_vmap_replace(
  ffi.Pointer<mlx_vector_array> res,
  mlx_vector_array inputs,
  mlx_vector_array s_inputs,
  mlx_vector_array s_outputs,
  ffi.Pointer<ffi.Int> in_axes,
  int in_axes_num,
  ffi.Pointer<ffi.Int> out_axes,
  int out_axes_num,
);