mlx_closure_custom_vmap_new_func function

  1. @Native<mlx_closure_custom_vmap Function(Pointer<NativeFunction<Int Function(Pointer<mlx_vector_array>, Pointer<mlx_vector_int>, mlx_vector_array, Pointer<Int>, Size)>>)>(ffi.Pointer<ffi.NativeFunction<ffi.Int Function(ffi.Pointer<mlx_vector_array>, ffi.Pointer<mlx_vector_int>, mlx_vector_array, ffi.Pointer<ffi.Int>, ffi.Size)>>)>()
mlx_closure_custom_vmap mlx_closure_custom_vmap_new_func(
  1. Pointer<NativeFunction<Int Function(Pointer<mlx_vector_array>, Pointer<mlx_vector_int>, mlx_vector_array, Pointer<Int>, Size)>> fun
)

Implementation

@ffi.Native<
  mlx_closure_custom_vmap Function(
    ffi.Pointer<
      ffi.NativeFunction<
        ffi.Int Function(
          ffi.Pointer<mlx_vector_array>,
          ffi.Pointer<mlx_vector_int>,
          mlx_vector_array,
          ffi.Pointer<ffi.Int>,
          ffi.Size,
        )
      >
    >,
  )
>()
external mlx_closure_custom_vmap mlx_closure_custom_vmap_new_func(
  ffi.Pointer<
    ffi.NativeFunction<
      ffi.Int Function(
        ffi.Pointer<mlx_vector_array>,
        ffi.Pointer<mlx_vector_int>,
        mlx_vector_array,
        ffi.Pointer<ffi.Int>,
        ffi.Size,
      )
    >
  >
  fun,
);