Skip to main content

recognize_full_attention

Function recognize_full_attention 

Source
pub fn recognize_full_attention(stmts: &[Stmt<'_>]) -> Option<FullAttentionSpec>
Expand description

Recognize Let C be full attention of Q over K and V with H heads.