zbj@%SSKrSSKrSSKr\RR \5r\RR\RR\S55(a6\RR\RR\S55 O5\RR\RR\S55 SSK J r J r J r SrSr\S-\S'\R("\5r"S S \ 5rg!\ar\r Sr Sr Sr\rSrCN3SrCff=f) Nz ../tools/symbolic_shape_infer.pyz../toolsz..)SymbolicShapeInferenceget_shape_from_type_protosympyT"_symbolic_shape_infer_import_errorFc^^\rSrSrS U4SjjrS S\\\4S\4SjjrSr Sr Sr S r U=r $) SymbolicShapeInferenceHelper c>[(d/[nUbS[U5;aSOSU<3n[SU35Ue[TU]X4XR5 XlSUlSUl0Ul g)Nrz1missing 'sympy' (install with: pip install sympy)z'failed to import symbolic_shape_infer: u0SymbolicShapeInferenceHelper is unavailable — F) _symbolic_shape_infer_availablerstr ImportErrorsuper__init__model_all_shapes_inferred_ is_inferred_dynamic_axis_mapping_) selfmodelverboseint_max auto_mergeguess_output_rankerrcause __class__s ڧ/tmp/claude-0/-home-danvics-docker-quiz/c1e0577a-e42c-4a3d-b1ea-3edd61103a4e/scratchpad/stt/lib/python3.13/site-packages/onnxruntime/transformers/shape_infer_helper.pyr%SymbolicShapeInferenceHelper.__init__!s|..4C?w#c(':D>sgF   PQVPWXY_b b .?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 _preprocessrrun_loggerdebug _infer_impl)rr r!counts rinfer"SymbolicShapeInferenceHelper.infer0s$///   !;!;?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 isinstancer rappendsymbolic_dims_rSymbol)rnodeidx sympy_shapeshapedims r_get_sympy_shape-SymbolicShapeInferenceHelper._get_sympy_shapeNs * 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_protor6ir7s rget_edge_shape+SymbolicShapeInferenceHelper.get_edge_shapeas(((( ~~ % 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)rrB Exception)rr? edge_otherr6 shape_others r compare_shape*SymbolicShapeInferenceHelper.compare_shapeysR((((##D)))*5 =K/OP P##r)rrrr)riTF))__name__ __module__ __qualname____firstlineno__rdictr intr*r8rBrH__static_attributes__ __classcell__)rs@rrr s7 8)$sCx.)C)<&0$$rr)loggingossyspathdirname__file__ file_pathexistsjoinr0symbolic_shape_inferrrrr rr __annotations__excobject getLoggerrKr&rrrrbs   GGOOH % 77>>"'',,y*LMNNHHOOBGGLLJ78HHOOBGGLLD12 -]]&*#=A& d(:A   8 $k$#9k$-# $ E&+#),& -sDD DD