torch_sys::c_generated

Function atg__native_multi_head_attention_out

source 
pub unsafe extern "C" fn atg__native_multi_head_attention_out(
    out__: *mut *mut C_tensor,
    out0_: *mut C_tensor,
    out1_: *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,
    need_weights_: c_int,
    average_attn_weights_: c_int,
    mask_type_v: i64,
    mask_type_null: i8,
)