atg__segment_reduce_backward

Function atg__segment_reduce_backward 

Source
pub unsafe extern "C" fn atg__segment_reduce_backward(
    out__: *mut *mut C_tensor,
    grad_: *mut C_tensor,
    output_: *mut C_tensor,
    data_: *mut C_tensor,
    reduce_ptr: *const u8,
    reduce_len: c_int,
    lengths_: *mut C_tensor,
    offsets_: *mut C_tensor,
    axis_: i64,
    initial_: *mut C_scalar,
)