h]SSKrSSKrSSKr\RR \5r\RR\RR\S55(a6\RR\RR\S55 O5\RR\RR\S55 SSK J r J r J r \R"\5r"SS\ 5rg)Nz ../tools/symbolic_shape_infer.pyz../toolsz..)SymbolicShapeInferenceget_shape_from_type_protosympyc^^\rSrSrS U4SjjrS S\\\4S\4SjjrSr Sr Sr S r U=r $) SymbolicShapeInferenceHelperc\>[TU]X4XR5 XlSUlSUl0Ulg)NF)super__init__model_all_shapes_inferred_ is_inferred_dynamic_axis_mapping_)selfmodelverboseint_max auto_mergeguess_output_rank __class__s e/var/www/fran/franai/venv/lib/python3.13/site-packages/onnxruntime/transformers/shape_infer_helper.pyr %SymbolicShapeInferenceHelper.__init__s0 .?I */!"'57"dynamic_axis_mappingmax_runscUceUR(aURU:Xa UR$XlURUR5 SnUR (aQ[ RSU35 UR5UlUS- nUS:aX2:aOUR (aMQSUlUR$)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. rzshape infer run T) rrr _preprocessr run_loggerdebug _infer_impl)rrrcounts rinfer"SymbolicShapeInferenceHelper.infers$///   !;!;?S!S,, ,%9" %ii LL+E73 4(,(8(8(:D % QJE!| 1 iii!(((rc/nURX5nU(aUHn[U[5(aXPR;a UR URU5 MGXPR ;a UR UR U5 MvUR [ R"USS95 MUceUR U5 M U$)zQOverride it to ensure shape inference by giving the actual value of dynamic axis.T)integer) _get_shape isinstancestrrappendsymbolic_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+rcbUR(deXR;a[S[U5-5 gURURn[ U5nUbM[ U5H>upE[U[5(dMXPR;dM-URUX4'M@ U$)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_printr*typer enumerater)r)redge type_protor1ir2s rget_edge_shape+SymbolicShapeInferenceHelper.get_edge_shapeOs(((( ~~ % 1CI= >^^D).. )*5  #E*c3''C3M3M,M#99#>EH+ rcUR(deURU5nURU5nUbUc [S5eX4:H$)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 r= Exception)rr: edge_otherr1 shape_others r compare_shape*SymbolicShapeInferenceHelper.compare_shapegsR((((##D)))*5 =K/OP P##r)r rrr )riTF))__name__ __module__ __qualname____firstlineno__r dictr*intr$r3r=rC__static_attributes__ __classcell__)rs@rrrs78)$sCx.)C)<&0$$rr)loggingossyspathdirname__file__ file_pathexistsjoinr+symbolic_shape_inferrrr getLoggerrFr rrrrZs   GGOOH % 77>>"'',,y*LMNNHHOOBGGLLJ78HHOOBGGLLD12YY   8 $c$#9c$r