torch_sys::c_generated

Function atg__efficient_attention_backward

source
pub unsafe extern "C" fn atg__efficient_attention_backward(
    out__: *mut *mut C_tensor,
    grad_out__: *mut C_tensor,
    query_: *mut C_tensor,
    key_: *mut C_tensor,
    value_: *mut C_tensor,
    bias_: *mut C_tensor,
    out_: *mut C_tensor,
    cu_seqlens_q_: *mut C_tensor,
    cu_seqlens_k_: *mut C_tensor,
    max_seqlen_q_: i64,
    max_seqlen_k_: i64,
    logsumexp_: *mut C_tensor,
    dropout_p_: f64,
    philox_seed_: *mut C_tensor,
    philox_offset_: *mut C_tensor,
    custom_mask_type_: i64,
    bias_requires_grad_: c_int,
    scale_v: f64,
    scale_null: i8,
    num_splits_key_v: i64,
    num_splits_key_null: i8,
    window_size_v: i64,
    window_size_null: i8,
    shared_storage_dqdkdv_: c_int,
)