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