ShapeInferencePass¶

class onnx_ir.passes.common.ShapeInferencePass(check_type=True, strict_mode=True, data_prop=True)[source]¶

This pass performs shape inference on the graph.

Parameters:
call(model)[source]¶

The main entry point for the pass.

Parameters:

model (Model)

Return type:

PassResult