ktj,UddlZddlZddlZejeZejejedr4ej ejedn3ej ejed ddl m Z m Z m Z dZdZedzed<n#e$rZeZ dZ dZ dZeZYdZ[ndZ[wwxYwejeZGd d e ZdS) Nz ../tools/symbolic_shape_infer.pyz../toolsz..)SymbolicShapeInferenceget_shape_from_type_protosympyT"_symbolic_shape_infer_import_errorFcTeZdZd fd Zddeeefdefd Zd Zd Z d Z xZ S)SymbolicShapeInferenceHelperrTFcts4t}|dt|vrdnd|}td||t ||||||_d|_d|_i|_ dS)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 i/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/onnxruntime/transformers/shape_infer_helper.pyrz%SymbolicShapeInferenceHelper.__init__!s. c4C?w#c((':':DCFsFF  XQVXXYY_b b *.?III */!"'57"""dynamic_axis_mappingmax_runscJ|J|jr|j|kr|jS||_||jd}|jrOt d|||_|dz }|dkr||krn|jOd|_|jS)aRun 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. Nrzshape infer run T) rrr _preprocessrrun_loggerdebug _infer_impl)rr r!counts rinferz"SymbolicShapeInferenceHelper.infer0s$///   -!;?S!S!S, ,%9" %%%i  LL3E33 4 4 4(,(8(8(:(:D % QJE!|| 1 1 i !((rcg}|||}|r|D]}t|tr~||jvr!||j|A||jvr!||j|k|t j|d|J|||S)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_shapez-SymbolicShapeInferenceHelper._get_sympy_shapeNs c**  , , ,c3'' ,d888#**4+Ec+JKKKK 333#**4+>s+CDDDD#**5<T+J+J+JKKKK???&&s++++rc8|jsJ||jvr!tdt|zdS|j|j}t |}|Ct |D]3\}}t|tr||jvr|j|||<4|S)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_protor5ir6s rget_edge_shapez+SymbolicShapeInferenceHelper.get_edge_shapeas(((( t~ % % 1CII= > > >4^D). )*55  #E** ? ?3c3''?C43M,M,M#9#>E!H rc|jsJ||}||}||td||kS)a*Compare 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 Nz1At least one shape is missed for edges to compare)rr@ Exception)rr= edge_otherr5 shape_others r compare_shapez*SymbolicShapeInferenceHelper.compare_shapeys]((((##D))))*55 =K/OPP P ##r)rr TF)r) __name__ __module__ __qualname__rdictr intr*r7r@rE __classcell__)rs@rrr s 8 8 8 8 8 8))$sCx.)C))))<&0$$$$$$$rr)loggingossyspathdirname__file__ file_pathexistsjoinr/symbolic_shape_inferrrrr rr __annotations__excobject getLoggerrFr&rrrr[s   GOOH % % 7>>"',,y*LMMNN3HOOBGLLJ778888HOOBGLLD11222 -]]]]]]]]]]&*#=A& d(:AAAA---# $ E&+#),&&&&&& -  8 $ $k$k$k$k$k$#9k$k$k$k$k$sCC3 C..C3