+ 9i]^RIt^RIt^RIt]PP ]4t]PP]PP]R44'd7]PP]PP]R44M5]PP]PP]R44^RI H t H t H t ]P!]4t!RR] 4tR#)Nz ../tools/symbolic_shape_infer.pyz../toolsz..)SymbolicShapeInferenceget_shape_from_type_protosympyc^aa]tRt^toRV3RlltR V3RlRlltRtRtRtRt Vt V;t #) SymbolicShapeInferenceHelperc^<\SV`W4WR4WnRVnRVn/VnR#)FN)super__init__model_all_shapes_inferred_ is_inferred_dynamic_axis_mapping_)selfmodelverboseint_max auto_mergeguess_output_rank __class__s&&&&&&i/var/www/html/photoedit/myenv/lib/python3.14/site-packages/onnxruntime/transformers/shape_infer_helper.pyr %SymbolicShapeInferenceHelper.__init__s0 .?I */!"'57"c<<V^8dQhRS[S[S[3,RS[/#)dynamic_axis_mappingmax_runs)dictstrint)format __classdict__s"r __annotate__)SymbolicShapeInferenceHelper.__annotate__s#))$sCx.)C)rc VfQhVP'dVPV8Xd VP#WnVPVP4^pVP 'dI\ PRV 24VP4VnV^, pV^8gKQW28gKYRVnVP#)ayRun shape inference, and try replace dynamic axis from string to integer when mapping is provided. Args: dynamic_axis_mapping (_type_): a dictionary with name of dynamic axis as key, like {"batch_size" : 4} max_runs (int, optional): limit maximum number of runs to avoid infinite loop. Defaults to 200. Returns: bool: whether all shapes has been inferred or not. zshape infer run T) r rr _preprocessr run_loggerdebug _infer_impl)rrrcounts&&& rinfer"SymbolicShapeInferenceHelper.infers$///    !;!;?S!S,, ,%9" %iii LL+E73 4(,(8(8(:D % QJE!| 1 (((rc .pVPW4pV'dVFp\V\4'dWPP9d%VP VPV,4KMWPP 9d%VP VP V,4KVP \ P!VRR74KVfQhVP V4K V#)zQOverride it to ensure shape inference by giving the actual value of dynamic axis.T)integer) _get_shape isinstancerrappendsymbolic_dims_rSymbol)rnodeidx sympy_shapeshapedims&&& r_get_sympy_shape-SymbolicShapeInferenceHelper._get_sympy_shape<s * c3''888#**4+E+Ec+JK 3 33#**4+>+>s+CD#**5<<T+JK?*?&&s+rc VP'gQhWP9d\R\V4,4R#VPV,Pp\ V4pVeT\ V4FDwrE\V\4'gKWPP9gK/VPV,W4&KF V#)zGet shape of an edge. Args: edge (str): name of edge Returns: Optional[List[int]]: the shape, or None if shape is unknown zCannot retrieve the shape of N) r known_vi_printrtyper enumerater0r)redge type_protor7ir8s&& rget_edge_shape+SymbolicShapeInferenceHelper.get_edge_shapeOs((((( ~~ % 1CI= >^^D).. )*5  #E*c3''C3M3M,M#99#>EH+ rc VP'gQhVPV4pVPV4pVeVf \R4hW48H#)zCompare shape of two edges. Args: edge (str): name of edge edge_other (str): name of another edge Raises: Exception: At least one shape is missed for edges to compare Returns: bool: whether the shape is same or not z1At least one shape is missed for edges to compare)r rC Exception)rr@ edge_otherr7 shape_others&&& r compare_shape*SymbolicShapeInferenceHelper.compare_shapegsU(((((##D)))*5 =K/OP P##r)r rr r )riTF)) __name__ __module__ __qualname____firstlineno__r r+r9rCrI__static_attributes____classdictcell__ __classcell__)rr!s@@rrrs(8))<&0$$rr)loggingossyspathdirname__file__ file_pathexistsjoinr1symbolic_shape_inferrrr getLoggerrLr'rrrr_s   GGOOH % 77>>"'',,y*LMNNHHOOBGGLLJ78HHOOBGGLLD12YY   8 $c$#9c$r