o
    h>                     @   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 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-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3 ddl4m5Z5m6Z7m8Z9m:Z; g dZ<de_=de _=de,_=de-_=de._=de/_=de0_=de1_=de2_=de3_=de5_=de;_=de9_=de7_=dZ>ej?Z@ejAdddddd ZBejCZDdddZEdddZF	 ejGZH	 ejIZJdS )    )_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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)ExportOptionsExportOutputExportOutputSerializerdynamo_exportOnnxExporterErrorenable_fake_modeOnnxRegistryDiagnosticOptions)is_onnxrt_backend_supported
OrtBackendOrtBackendOptionsOrtExecutionProvider)'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-   r.   r/   z
torch.onnxpytorchz1.12.0z2.0zuse `torch.onnx.export` instead)since
removed_ininstructionsc                  O   s   t j| i |S )N)r   _export)argskwargs r<   I/var/www/html/ai/venv/lib/python3.10/site-packages/torch/onnx/__init__.pyr9      s   r9   returnNc                   C      t d dS )zEnables ONNX logging.TNr   _jit_set_onnx_log_enabledr<   r<   r<   r=   r4         r4   c                   C   r?   )zDisables ONNX logging.FNr@   r<   r<   r<   r=   r3      rB   r3   )r>   N)Ktorchr   torch._Cr   _C_onnxtorch._C._onnxr   r   r   r    r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   _exporter_statesr   r   _type_utilsr   r   r   r   r    r!   r"   r#   r$   r%   r&   _internal.exporterr'   r(   r)   r*   r+   r,   r-   r.   _internal.onnxruntimer/   r0   _OrtBackendr1   _OrtBackendOptionsr2   _OrtExecutionProvider__all__
__module__producer_namePRODUCER_VERSIONproducer_version
deprecatedr9   _jit_is_onnx_log_enabledis_onnx_log_enabledr4   r3   _jit_set_onnx_log_output_streamset_log_stream_jit_onnx_loglogr<   r<   r<   r=   <module>   sL    L,(2



