Skip to main content

float_tensors

Function float_tensors 

Source
pub fn float_tensors(rp: &RecognizedProgram) -> BTreeSet<String>
Expand description

Which tensors of a walked program hold FLOAT data, propagated forward from the declarations, each name as its last write leaves it.

★ AN ACTIVATION HAS NO FILL TO READ. A backend decides between an integer and a float datapath from the declared fill of its operands — right for a declared tensor, blind for one another op produces. The walk knows: a sugar that computes an exponential or a normalization is float by construction, and everything else is float exactly when something feeding it is.