Skip to main content

InferenceFn

Type Alias InferenceFn 

Source
pub type InferenceFn = fn(&mut InferenceContext<'_>) -> Result<(), ShapeInferError>;
Expand description

An operator inference rule: reads inputs from the InferenceContext and sets its outputs’ types (and, where applicable, shape-data).