Skip to main content

unit_output

logicaffeine_compile::tensor_graph::fuse

Function unit_output 

Source 
pub fn unit_output(op: &Op) -> Option<(usize, usize)>
Expand description

The units an op writes its result in, (units, width), when it can write them as a part of every unit of a wider tensor: a norm, a rotation, a per-head product.