B
    kn¦dS  ã               @   s.   d dl mZ ddd„Zdd	d
„Zddd„ZdS )é    )Úloggingé	   FÚONNXTNc       	   
   C   s$   ddl m} || ||||||||ƒ	S )Nr   )Úexport_onnx)Úpaddle2onnx.legacy.convertr   )	Zpaddle_graphÚ	save_fileÚopset_versionÚenable_onnx_checkerÚoperator_export_typeÚverboseÚauto_update_opsetZoutput_namesr   © r   ú7/tmp/pip-unpacked-wheel-qdbcl8o8/paddle2onnx/convert.pyr      s    
r   c             K   sX  d|kr.|d s.ddl m} || |||f|ŽS dd l}dd l}dd l}|j |¡d }	|j |	d¡}
|j |
d¡}|j |
d¡}|j |
¡rÞ|j 	|
¡r²t
 d |
¡¡ | |
¡ |j 	|¡rÈ| |¡ |j 	|¡rÞ| |¡ |j | |j |
d¡|¡ t
 d	 |
¡¡ |j 	|¡sd
}|d kr4| ||||¡S | ||||¡ t
 d |¡¡ d S )NZenable_dev_versionr   )Údygraph2onnxZ!paddle_model_static_onnx_temp_dirzmodel.pdmodelzmodel.pdiparamszFile {} exists, will remove it.Úmodelz&Static PaddlePaddle model saved in {}.Ú zONNX model saved in {}.)r   r   ÚosÚpaddle2onnxÚpaddleÚpathÚsplitÚjoinÚexistsÚisfiler   ÚinfoÚformatÚremoveZjitÚsaveÚexport)Zlayerr   Z
input_specr   Úconfigsr   r   r   r   ÚdirnameZpaddle_model_dirZ
model_fileZparams_filer   r   r   r       s>    




r   c	          
   K   s(   ddl m}
 |
| ||||||||f	|	ŽS )Nr   )Úprogram2onnx)r   r!   )ÚprogramÚscoper   Zfeed_var_namesZtarget_varsr   r	   r
   r   r   r!   r   r   r   r!   F   s    
r!   )r   Fr   FTN)Nr   )NNr   Fr   T)Zpaddle2onnx.utilsr   r   r   r!   r   r   r   r   Ú<module>   s        

)     