Searched defs:debug_attn_mask (Results 1 – 2 of 2) sorted by relevance
464 auto debug_attn_mask = at::empty({batch_size, num_heads, max_seqlen_q, max_seqlen_kv}, in custom_scaled_dot_product_fused_attention_overrideable() local
879 philox_seed, philox_offset, debug_attn_mask; in _flash_attention_forward() local