Function atg__triton_multi_head_attention_out

pub unsafe extern "C" fn atg__triton_multi_head_attention_out(
    out__: *mut *mut C_tensor,
    out_: *mut C_tensor,
    query_: *mut C_tensor,
    key_: *mut C_tensor,
    value_: *mut C_tensor,
    embed_dim_: i64,
    num_head_: i64,
    qkv_weight_: *mut C_tensor,
    qkv_bias_: *mut C_tensor,
    proj_weight_: *mut C_tensor,
    proj_bias_: *mut C_tensor,
    mask_: *mut C_tensor,
)