Function mlx_number_of_elements 

pub unsafe extern "C" fn mlx_number_of_elements(
    res: *mut mlx_array,
    a: mlx_array,
    axes: *const c_int,
    axes_num: usize,
    inverted: bool,
    dtype: mlx_dtype,
    s: mlx_stream,
) -> c_int