torch_sys::c_generated

Function atg_segment_reduce_out

source
pub unsafe extern "C" fn atg_segment_reduce_out(
    out__: *mut *mut C_tensor,
    out_: *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,
)