pub fn unit_input(op: &Op) -> Option<(String, usize, usize)>Expand description
The input an op reads unit by unit and its units, (x, units, width), for an op that can read them as a part of
every unit of a wider tensor: a norm (not of what it adds), a rotation, a per-head product — none reading in place
already.