o
    ;ήct	                     @   sN   d dl Z e jjZd dlmZ ddejddddddddddddddfddZdS )    N)VersionTFc                 C   s   t tjt dkr tjj| |||||||||	|||||d d S tjjdi d| d|d|d|d|d|d	|d
|d|d|	d|
d|d|d|d|d|d|d|d| d S )Nz1.11.0)modelargsfexport_paramsverbosetraininginput_namesoutput_namesoperator_export_typeopset_versiondo_constant_foldingdynamic_axeskeep_initializers_as_inputscustom_opsetsexport_modules_as_functionsr   r   r   r   r   r   r	   r
   r   r   _retain_param_namer   example_outputsstrip_doc_stringr   r   r   enable_onnx_checkeruse_external_data_format )r   torch__version__onnxexport)r   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   X/tmp/pip-target-vg8gfxp4/lib/python/onnxruntime/transformers/torch_onnx_export_helper.pytorch_onnx_export   sr   
	
r   )r   r   TrainingModepackaging.versionr   EVALr   r   r   r   r   <module>   s*   