mlx_ones function

int mlx_ones(
  1. Pointer<mlx_array> res,
  2. Pointer<Int> shape,
  3. int shape_num,
  4. mlx_dtype_ dtype,
  5. mlx_stream s,
)

Implementation

int mlx_ones(
  ffi.Pointer<mlx_array> res,
  ffi.Pointer<ffi.Int> shape,
  int shape_num,
  mlx_dtype_ dtype,
  mlx_stream s,
) => _mlx_ones(res, shape, shape_num, dtype.value, s);