torch_sys::c_generated

Function atg__triton_scaled_dot_attention_out

source
pub unsafe extern "C" fn atg__triton_scaled_dot_attention_out(
    out__: *mut *mut C_tensor,
    out_: *mut C_tensor,
    q_: *mut C_tensor,
    k_: *mut C_tensor,
    v_: *mut C_tensor,
    dropout_p_: f64,
)