Function torch_sys::c_generated::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
)