pub fn recognize_unary_map(stmts: &[Stmt<'_>]) -> Option<UnaryMapSpec>Expand description
Recognize the one-loop unary map Let Y be silu/sigmoid/gelu of X anywhere in the statement list.
pub fn recognize_unary_map(stmts: &[Stmt<'_>]) -> Option<UnaryMapSpec>Recognize the one-loop unary map Let Y be silu/sigmoid/gelu of X anywhere in the statement list.