torch_sys::c_generated

Function atg__masked_softmax

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