rewrite

onnxscript.rewriter.rewrite(model: _ModelProtoOrIr, pattern_rewrite_rules: Sequence[RewriteRule] | RewriteRuleSet | None = None) → _ModelProtoOrIr[source]

Rewrite the model using the provided pattern rewrite rules.

Unused nodes, functions, and opsets will be removed after the rewrite.

Parameters:

  • model – The model to be rewritten. Can be an ONNX ModelProto or an ir.Model.

  • pattern_rewrite_rules – A sequence of pattern rewrite rules or a RewriteRuleSet. If not provided, default rules will be applied. If empty, no rules will be applied and the original model will be returned.

Returns:

The rewritten model as the same type as the input model.