pub unsafe extern "C" fn atg_segment_reduce(
    out__: *mut *mut C_tensor,
    data_: *mut C_tensor,
    reduce_ptr: *const u8,
    reduce_len: c_int,
    lengths_: *mut C_tensor,
    indices_: *mut C_tensor,
    offsets_: *mut C_tensor,
    axis_: i64,
    unsafe_: c_int,
    initial_: *mut C_scalar
)