mlx_number_of_elements function

int mlx_number_of_elements(
  1. Pointer<mlx_array> res,
  2. mlx_array a,
  3. Pointer<Int> axes,
  4. int axes_num,
  5. bool inverted,
  6. mlx_dtype_ dtype,
  7. mlx_stream s,
)

Implementation

int mlx_number_of_elements(
  ffi.Pointer<mlx_array> res,
  mlx_array a,
  ffi.Pointer<ffi.Int> axes,
  int axes_num,
  bool inverted,
  mlx_dtype_ dtype,
  mlx_stream s,
) => _mlx_number_of_elements(res, a, axes, axes_num, inverted, dtype.value, s);