
    h                     F    S SK r S SKrS SKrS SKrS r\S:X  a  \" 5         gg)    Nc                     [         R                  " [        R                  R	                  [
        5      SS9n U R                  S[        R                  SS9  U R                  S[        R                  SS9  U R                  5       n[        R                  " [        UR                  R                  SS	95      5      n[        S
5        [        R                   " U[        UR"                  R                  5       5      5        g )Nz^Update a QDQ format ONNX model to ensure optimal performance when executed using ONNX Runtime.)descriptioninput_modelz%Provide path to ONNX model to update.)typehelpoutput_modelz,Provide path to write updated ONNX model to.T)strictzfAs of ORT 1.15, the fixing up of DQ nodes with multiple consumers is done by an ORT graph transformer.)argparseArgumentParserospathbasename__file__add_argumentpathlibPath
parse_argsonnxloadstrr   resolveprintsaver   )parserargsmodels      j/var/www/fran/franai/venv/lib/python3.13/site-packages/onnxruntime/tools/qdq_helpers/optimize_qdq_model.pyoptimize_qdq_modelr      s    $$
"tF
 GLL?fg
W\\@noDIIc$**22$2?@AE 

rs 	IIeS**22456    __main__)r
   r   r   r   r   __name__ r   r   <module>r#      s-   
  	  70 z r   