o gu@sddlZddlZddlZddlmZejeZej ej edr/ej ej edn ej ej edddl m Z mZmZeeZGddde ZdS) N)Dictz ../tools/symbolic_shape_infer.pyz../toolsz..)SymbolicShapeInferenceget_shape_from_type_protosympycsReZdZdfdd Zddeeefd efd d Zd d ZddZ ddZ Z S)SymbolicShapeInferenceHelperrTFcs.t||||||_d|_d|_i|_dS)NF)super__init__model_all_shapes_inferred_ is_inferred_dynamic_axis_mapping_)selfmodelverboseint_max auto_mergeguess_output_rank __class__b/var/www/visachat/venv/lib/python3.10/site-packages/onnxruntime/transformers/shape_infer_helper.pyr s  z%SymbolicShapeInferenceHelper.__init__dynamic_axis_mappingmax_runscCs|dusJ|jr|j|kr|jS||_||jd}|jr|jsJ||}||}|dus|durtd||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)r r7 Exception)rr4 edge_otherr- shape_otherrrr compare_shapehs  z*SymbolicShapeInferenceHelper.compare_shape)rrTF)r) __name__ __module__ __qualname__r rr&intr"r/r7r; __classcell__rrrrrs r)loggingossystypingrpathdirname__file__ file_pathexistsjoinr'symbolic_shape_inferrrr getLoggerr<rrrrrrs