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