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.
pub fn recognize_full_attention(stmts: &[Stmt<'_>]) -> Option<FullAttentionSpec>Recognize Let C be full attention of Q over K and V with H heads.