Function torch_sys::c_generated::atg_scaled_dot_product_attention
source · pub unsafe extern "C" fn atg_scaled_dot_product_attention(
out__: *mut *mut C_tensor,
query_: *mut C_tensor,
key_: *mut C_tensor,
value_: *mut C_tensor,
attn_mask_: *mut C_tensor,
dropout_p_: f64,
is_causal_: c_int,
scale_v: f64,
scale_null: i8
)