Skip to main content

recognize_argmax

logicaffeine_compile::tensor_recognize::qwen

Function recognize_argmax 

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

Recognize Let t be argmax of X.