TiSrSSKJr SS/rSSKrSSKrSSKrSSKJ r \R"\ 5r S Sjr "SS\RR5rS S S S .S S jjrg)z0Shape inference pass using onnx.shape_inference.) annotationsShapeInferencePass infer_shapesN) _c_api_utilsc[RRU5nSn[UR 5UR 55GHupE[R R U5n[R R U5nUR5HupX;a~Xxn U RU R:wa U RbU RU lSnU RU R:wa$U RbU RU l SnMMM[RSXR5 M GM U$)aMerge the shape inferred model with the original model. Args: model: The original IR model. inferred_proto: The ONNX model with shapes and types inferred. Returns: A tuple containing the modified model and a boolean indicating whether the model was modified. FTz'Value %s not found in inferred graph %s) irserdedeserialize_modelzipgraphs conveniencecreate_value_mappingitemsshapedtypeloggerwarningname) modelinferred_protoinferred_modelmodifiedoriginal_graphinferred_graphoriginal_valuesinferred_valuesrvalueinferred_values d/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx_ir/passes/common/shape_inference.py _merge_funcr sXX//?NH*-ellnn>S>S>U*V&..==nM..==nM*002KD&!0!6;;."6"66>;O;O;["0"6"6EK#H;;."6"66>;O;O;["0"6"6EK#H<\6=tEXEX3+W OcN^\rSrSrSrSSU4SjjjrSSjrSrU=r$) r5z0This pass performs shape inference on the graph.cF>[TU]5 XlX lX0lg)a#Initialize the shape inference pass. If inference fails, the model is left unchanged. Args: check_type: If True, check the types of the inputs and outputs. strict_mode: If True, use strict mode for shape inference. data_prop: If True, use data propagation for shape inference. N)super__init__ check_type strict_mode data_prop)selfr'r(r) __class__s rr&ShapeInferencePass.__init__8s  $&"r!c(^SU4Sjjn[R"X!5n[X5n[ R RXS9$![a>n[R SUS9 [ R RUS5sSnA$SnAff=f)Nc>[RRUTRTRTR S9$)Nr'r(r))onnxshape_inferencerr'r(r))protor*s rpartial_infer_shapes5ShapeInferencePass.call..partial_infer_shapesJs;''44?? ,,.. 5 r!z3Shape inference failed: %s. Model is left unchanged)exc_infoF)r)r2onnx.ModelProtoreturnr6) r call_onnx_api Exceptionrrrpasses PassResultr )r*rr3inferred_model_protoers` rcallShapeInferencePass.callIs~  6#/#=#=>R#Z u;yy##E#==  6 NNP[\N ]99''u5 5 6sA B3B B B)r'r)r()TTT)r'boolr(r@r)r@r7None)rir.Modelr7zir.passes.PassResult) __name__ __module__ __qualname____firstlineno____doc__r&r>__static_attributes__ __classcell__)r+s@rrr5sA:TX##48#LP# ##">>r!Tr/c6[XUS9"U5R$)a\Perform shape inference on the model. Args: model: The model to perform shape inference on. check_type: If True, check the types of the inputs and outputs. strict_mode: If True, use strict mode for shape inference. data_prop: If True, use data propagation for shape inference. Returns: The model with shape inference applied. r/)rr)rr'r(r)s rrr\s#$ )   Ur!)rrBrr6r7r@) rrBr'r@r(r@r)r@r7rB)rG __future__r__all__loggingr0onnx_irronnx_ir.passes.commonr getLoggerrCrr r: InPlacePassrrr!rrSs7"   .   8 $>$>..$>T       r!