Ti"VSSKJr SSKJr /SQrSSKrSSKJs Jr SSK J s J r SSKJr SSK JrJr SSK Jr SSKJr \"S \R,\R.5rSS S \ R2\ R4S S .SS jjjrSS jrSSjrSSjrSSjrg)) annotations)TypeVar)basic_constant_propagationfold_constants_irfold_constantsFOLDED_FROM_KEYinline optimize_iroptimizeremove_unused_nodesN)ir)rr)r)r _ModelProtoOrIrT)onnx_shape_inferencestop_if_no_changeinput_size_limitoutput_size_limitr c B[U[R5(a[UUUUUUUS9 U$[U[R 5(de[R RU5n[UUUUUUUS9 [R RU5nU$)aPOptimizes a model. Args: model: The model to be optimized. num_iterations: Number of times the optimization loop is repeated. onnx_shape_inference: Applies node-level shape-inference as part of optimization input_size_limit: Will not apply constant folding to ops with any input of size greater than this. Does not apply to special ops like Shape() and Size(). output_size_limit: Will not rewrite any foldable-op into a Constant op if the size of the output tensor is greater than this. stop_if_no_change: Stop the optimization loop if no change is detected in an iteration. inline: If True, inlines all functions in the model. Returns: The optimized model. If the input was a ModelProto, the output will also be a ModelProto. If the input was an ir.Model, the output will also be an ir.Model. )num_iterationsrrrrr ) isinstancer Modelr onnx ModelProtoserdedeserialize_modelserialize_model) modelrrrrrr model_ir new_protos \/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnxscript/optimizer/__init__.pyr r s6%""  )!5/-/  %111188--e4 )!5/-/ HH,,X6 c^UR(a[R"5"U5 gg)z5Inline all function calls (recursively) in the model.N) functions common_passes InlinePass)rs rr r Ws   "5)r c[U[R5(a[R"U/UQ70UD6$[U[ R 5(deUn[RRU5n[R"U/UQ70UD6n[RRU5nUR5 URU5 U$)z#Fold constants in a model in place.) rr rconstant_foldingrrrrrrClearCopyFrom)rargskwargs model_protoresultrs rrr]s%""..uFtFvFF%1111 **;7!00HHHHH,,U3 Y' r c[U[R5(a[R"5"U5 g[R R U5n[R"5"U5Rn[R RU5nUR5 URU5 g)z*Removes unused nodes from a model inplace.N) rr rr#RemoveUnusedNodesPassrrrrr'r(rrrs rr r osz%""++-e488--e4 668BHHHH,,X6   y!r c[U[R5(a[R"5"U5 g[R R U5n[R"5"U5Rn[R RU5nUR5 URU5 g)z.Removes unused functions from a model inplace.N) rr rr#RemoveUnusedFunctionsPassrrrrr'r(r/s rremove_unused_functionsr2{sz%""//1%888--e4 ::rGs#  --AA^V7+T__bhhG 6"&",SS-UU6 66 6  6  66 66r*  %)$ " "r