o åe¤ ã @sŒdZddlZddlmmZddlmZddlmZddeee fde de d e d ef d d „Z dde de de de d e d df dd„Z ejZdS)zIonnx shape inference. Shape inference is not guaranteed to be complete. éN)Ú ModelProto)ÚUnionFÚmodelÚ check_typeÚ strict_modeÚ data_propÚreturncCsbt|ttfƒrt|tƒr|n| ¡}t ||||¡}t |¡St|tƒr(t dƒ‚t d  t |ƒ¡ƒ‚)aApply shape inference to the provided ModelProto. Inferred shapes are added to the value_info field of the graph. If the inferred values conflict with values already provided in the graph, that means that the provided values are invalid (or there is a bug in shape inference), and the result is unspecified. Arguments: model (Union[ModelProto, bytes], bool, bool, bool) -> ModelProto check_type (bool): Checks the type-equality for input and output strict_mode (bool): Stricter shape inference, it will throw errors if any; Otherwise, simply stop if any error data_prop (bool): Enables data propagation for limited operators to perform shape computation Returns: (ModelProto) model with inferred shape information zhinfer_shapes only accepts ModelProto or bytes,you can use infer_shapes_path for the model path (String).zAinfer_shapes only accepts ModelProto or bytes, incorrect type: {}) Ú isinstancerÚbytesZSerializeToStringÚCÚ infer_shapesÚonnxZload_from_stringÚstrÚ TypeErrorÚformatÚtype)rrrrZ model_strZinferred_model_str©rúDD:\Projects\ConvertPro\env\Lib\site-packages\onnx/shape_inference.pyr s   ÿr ÚÚ model_pathÚ output_pathcCsPt|tƒr tdƒ‚t|tƒr|dkr|}t |||||¡dStd t|ƒ¡ƒ‚)z· Take model path for shape_inference same as infer_shape; it support >2GB models Directly output the inferred model to the output_path; Default is the original model path z_infer_shapes_path only accepts model Path (String),you can use infer_shapes for the ModelProto.rzFinfer_shapes_path only accepts model path (String), incorrect type: {}N)r rrrr Úinfer_shapes_pathrr)rrrrrrrrr,s   ÿr)FFF)rFFF)Ú__doc__r Z'onnx.onnx_cpp2py_export.shape_inferenceZonnx_cpp2py_exportZshape_inferencer rÚtypingrr Úboolr rrZInferenceErrorrrrrÚs  ($