Skip to main content

register_shape_inference

Function register_shape_inference 

Source
pub fn register_shape_inference(
    registry: &mut InferenceRegistry,
    domain: &str,
    op_type: &str,
    min_opset: u64,
    handler: InferenceFn,
)
Expand description

Register an opset-aware shape-inference function on registry.

The underlying engine uses function pointers receiving an InferenceContext, rather than a global handler trait. A rule registered at min_opset applies until superseded by a registration at a higher opset.