torch_sys::c_generated

Function atg__masked_softmax_out

source
pub unsafe extern "C" fn atg__masked_softmax_out(
    out__: *mut *mut C_tensor,
    out_: *mut C_tensor,
    self_: *mut C_tensor,
    mask_: *mut C_tensor,
    dim_v: i64,
    dim_null: i8,
    mask_type_v: i64,
    mask_type_null: i8,
)