U
    <c>                     @   sn  d Z ddlmZ ddlmZ ddlmZmZm	Z	m
Z
 ddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ ddlmZmZ ddlm Z  dd	lm!Z! dd
lm"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z* dddddddddddddddddddddd d!d"d#d$d%d&d'gZ+d(e_,d(e _,d)Z-ej.Z/ej0d*d+d,d-d.d/ Z1ej2Z3d0d1d2d&Z4d0d1d3d%Z5ej6Z7ej8Z9d0S )4zONNX exporter.    )_C)_onnx)_CAFFE2_ATEN_FALLBACKOperatorExportTypesTensorProtoDataTypeTrainingMode   )_deprecationerrorssymbolic_caffe2symbolic_helpersymbolic_opset7symbolic_opset8symbolic_opset9symbolic_opset10symbolic_opset11symbolic_opset12symbolic_opset13symbolic_opset14symbolic_opset15symbolic_opset16symbolic_opset17utils)ExportTypesSymbolicContext)JitScalarType)CheckerError)	_optimize_graph_run_symbolic_function_run_symbolic_methodexportexport_to_pretty_stringis_in_onnx_exportregister_custom_op_symbolicselect_model_mode_for_exportunregister_custom_op_symbolicr   r   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r$   r#   r%   disable_log
enable_logr   z
torch.onnxZpytorchz1.12.0z1.14zuse `torch.onnx.export` instead)ZsinceZ
removed_inZinstructionsc                  O   s   t j| |S )N)r   _export)argskwargs r+   7/tmp/pip-unpacked-wheel-gikjz4vx/torch/onnx/__init__.pyr(   \   s    r(   N)returnc                   C   s   t d dS )zEnables ONNX logging.TNr   Z_jit_set_onnx_log_enabledr+   r+   r+   r,   r'   i   s    c                   C   s   t d dS )zDisables ONNX logging.FNr.   r+   r+   r+   r,   r&   n   s    ):__doc__Ztorchr   Ztorch._Cr   Z_C_onnxZtorch._C._onnxr   r   r   r    r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   Z_exporter_statesr   r   Z_type_utilsr   r   r   r   r   r    r!   r"   r#   r$   r%   __all__
__module__Zproducer_nameZPRODUCER_VERSIONZproducer_version
deprecatedr(   Z_jit_is_onnx_log_enabledZis_onnx_log_enabledr'   r&   Z_jit_set_onnx_log_output_streamZset_log_streamZ_jit_onnx_loglogr+   r+   r+   r,   <module>   sj   H,&  
	