U
    Mh                  +   @   s.  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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
l-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7 ddl8m9Z9m: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(d)d*d+d,d-d.d/d0d1d2d3d4d5d6g+Z@d7e_Ad7e"_Ad7e/_Ad7e0_Ad7e1_Ad7e2_Ad7e6_Ad7e3_Ad7e4_Ad7e7_Ad7e5_Ad7e._Ad7e9_Ad7e?_Ad7e=_Ad7e;_Ad8ZBejCZDejEd9d:d;d<d=d> ZFejGZHd?d@dAd*ZId?d@dBd)ZJejKZLejMZNd?S )C    )_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symbolic_opset18symbolic_opset19symbolic_opset20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_symbolic)
DiagnosticOptionsExportOptionsONNXProgramONNXProgramSerializerONNXRuntimeOptionsInvalidExportOptionsErrorOnnxExporterErrorOnnxRegistrydynamo_exportenable_fake_mode)is_onnxrt_backend_supported
OrtBackendOrtBackendOptionsOrtExecutionProviderr   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%   r'   r&   r(   disable_log
enable_logr   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   z
torch.onnxZpytorchz1.12.0z2.0zuse `torch.onnx.export` instead)ZsinceZ
removed_inZinstructionsc                  O   s   t j| |S )N)r   _export)argskwargs r<   E/var/www/html/venv/lib/python3.8/site-packages/torch/onnx/__init__.pyr9      s    r9   N)returnc                   C   s   t d dS )zEnables ONNX logging.TNr   Z_jit_set_onnx_log_enabledr<   r<   r<   r=   r8      s    c                   C   s   t d dS )zDisables ONNX logging.FNr?   r<   r<   r<   r=   r7      s    )OZ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   r   r   r   Z_exporter_statesr   r   Z_type_utilsr   r   r    r!   r"   r#   r$   r%   r&   r'   r(   Z_internal.exporterr)   r*   r+   r,   r-   r.   r/   r0   r1   r2   Z_internal.onnxruntimer3   r4   Z_OrtBackendr5   Z_OrtBackendOptionsr6   Z_OrtExecutionProvider__all__
__module__Zproducer_nameZPRODUCER_VERSIONZproducer_version
deprecatedr9   Z_jit_is_onnx_log_enabledZis_onnx_log_enabledr8   r7   Z_jit_set_onnx_log_output_streamZset_log_streamZ_jit_onnx_loglogr<   r<   r<   r=   <module>   s   T,0	6  
	