
    h                    Z    S SK Jr  S SKrS SKrS SKrSSKJrJr  S r\	S:X  a  \" 5         gg)    )annotationsN   )get_optimization_leveloptimize_modelc            	        [         R                  " [        R                  R	                  [
        5       S[        R                   3SS9n U R                  SS/ SQSS9  U R                  S	/ S
Q[        SSSS9  U R                  S[        R                  SS9  U R                  S[        R                  SS9  U R                  5       nUR                  S:X  a  SnO8UR                  S:X  a  SnO%UR                  S:X  a  SnOUR                  S:X  a  Sn[        UR                  UR                   [#        UR$                  5      W5        g )N:a7  
                                     Optimize an ONNX model using ONNX Runtime to the specified level.
                                     See https://onnxruntime.ai/docs/performance/model-optimizations/graph-optimizations.html for more
                                     details of the optimization levels.)descriptionz--opt_levelbasic)disabler
   extendedlayoutallzOptimization level to use.)defaultchoiceshelpz--log_level)debuginfowarningerrorFr   zLog level. Defaults to Error so we don't get output about unused initializers being removed. Warning or Info may be desirable in some scenarios.)r   typerequiredr   r   input_modelz%Provide path to ONNX model to update.)r   r   output_modelz.Provide path to write optimized ONNX model to.   r   r   r   r   r      )argparseArgumentParserospathbasename__file__optimize_model_helper__name__add_argumentstrpathlibPath
parse_args	log_levelr   r   r   r   	opt_level)parserargsr)   s      _/var/www/fran/franai/venv/lib/python3.13/site-packages/onnxruntime/tools/optimize_onnx_model.pyr"   r"      s6   $$77H%
&a(=(F(F'GHLF A)	   5M   GLL?fg
W\\@pqD~~ 		7	"		6	!		9	$	4##T%6%68Nt~~8^`ij    __main__)

__future__r   r   r   r&   onnx_model_utilsr   r   r"   r#    r.   r-   <module>r3      s2    #  	  D'kT z r.   