Tiq%SSKJr SSKJrJrJr /SQrSSKrSSKJ s J r SSK J r SSKJr SSKJrJrJrJr SSKJrJrJrJrJrJr SS KJrJrJrJ r J!r!J"r"J#r#J$r$J%r%J&r&J'r' \"S \RP\ RR5r*/\%RVQ\RVQ\RVQ\ RVQ\$RVQ\#RVQ\RX"5Q\&RVQ\"RVQ\!RVQ\'RVQ7r-S \.S '"S S\ RR^5r0SSSjjr1g)) annotations)SequenceTypeVarUnion) merge_metadatapatternrewrite RewritePass MatchResult MatchContext RewriteRuleRewriteRuleClassBaseRewriteRuleSetRewriterContextMatchingTracer MatchStatus RULE_NAME_TAGN)ir)r)r rr r)rrr rrr) _basic_rules_broadcast_to_matmul_cast_constant_of_shape_collapse_slices_fuse_batchnorm_fuse_pad_into_conv_fuse_relus_clips_min_max_to_clip_no_op_redundant_scatter_nd_remove_optional_bias_ModelProtoOrIrztuple[pattern.RewriteRule, ...]_DEFAULT_REWRITE_RULESc>^\rSrSrSU4SjjrSSjrSrU=r$)r Cc>[TU]5 [U[5(a(U(d [ S5e[ R "U5n[U[ R 5(deXlg)Nzrules must not be empty)super__init__ isinstancer ValueErrorrrrules)selfr) __class__s [/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnxscript/rewriter/__init__.pyr&RewritePass.__init__DsZ  eX & & !:;;**51E%!7!78888-2 cURRU5nU(a[SUS35 [RR U[ U55$)NzApplied z" of general pattern rewrite rules.)r)apply_to_modelprintrpasses PassResultbool)r*modelcounts r,callRewritePass.callRsF ))%0  HUG#EF Gyy##E4;77r.)r))r)z6Sequence[pattern.RewriteRule] | pattern.RewriteRuleSetreturnNone)r5zir.Modelr9zir.passes.PassResult)__name__ __module__ __qualname____firstlineno__r&r7__static_attributes__ __classcell__)r+s@r,r r Cs$ 3E 3  388r.r cUc[nO U(dU$[U[R5(a"[R R U5nSnOUnSn[RR[U5[R"5[R"5[R"545nU"U5RnU(a[R RU5$U$)aRewrite the model using the provided pattern rewrite rules. Unused nodes, functions, and opsets will be removed after the rewrite. Args: 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. TF)r!r'onnx ModelProtorserdedeserialize_modelr2 PassManagerr common_passesRemoveUnusedNodesPassRemoveUnusedFunctionsPassRemoveUnusedOpsetsPassr5serialize_model)r5pattern_rewrite_rulesmodel_irproto rewrite_passs r,r r Ys$$ 6 " %))88--e499(( - .  / / 1  3 3 5  0 0 2  LH%++H xx''11 Or.)N)r5r rLzCUnion[Sequence[pattern.RewriteRule], pattern.RewriteRuleSet] | Noner9r )2 __future__rtypingrrr__all__rBonnx_ir.passes.commonr2commonrG onnxscriptronnxscript.rewriterronnxscript.rewriter._basicsr rr r!onnxscript.rewriter._rewrite_rulerrr rrr onnxscript.rewriter.rules.commonrrrrrrrrrrrrCModelr r)basic_optimization_rulesr!__annotations__ InPlacePassr r r.r,r_s^#++   --'^^    +T__bhhG ; \\ ; ;"" ; ;  ;  ;**, ;   ; ; ;   ;7 8"))''82) ) ) )r.