o
    ;ήcn                     @   sJ   d dl Z d dlZd dlZd dlZddlmZ dd Zedkr#e  dS dS )    N   )$fix_dq_nodes_with_multiple_consumersc                  C   s|   t jtjtdd} | jdtjdd | jdtjdd | 	 }t
t|jjdd	}t| t
|t|j  d S )
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)strict)argparseArgumentParserospathbasename__file__add_argumentpathlibPath
parse_argsonnxloadstrr   resolver   saver   )parserargsmodel r   W/tmp/pip-target-vg8gfxp4/lib/python/onnxruntime/tools/qdq_helpers/optimize_qdq_model.pyoptimize_qdq_model   s   
r   __main__)r
   r   r   r   qdq_model_utilsr   r   __name__r   r   r   r   <module>   s   
