U
    •Ê¹j>  ã                '   @   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; 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/d0d1d2g'Z<d3e_=d3e _=d3e,_=d3e-_=d3e._=d3e/_=d3e0_=d3e1_=d3e2_=d3e3_=d3e5_=d3e;_=d3e9_=d3e7_=d4Z>ej?Z@ejAd5d6d7d8�d9d:„ ƒZBejCZDd;d<œd=d(„ZEd;d<œd>d'„ZFejGZHe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Ú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&   Údisable_logÚ
enable_logr   r'   r(   r)   r*   r+   r,   r-   r.   r/   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© r8   úR/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/onnx/__init__.pyr5   …   s    r5   N)Úreturnc                   C   s   t  d¡ dS )zEnables ONNX logging.TN©r   Z_jit_set_onnx_log_enabledr8   r8   r8   r9   r4   ’   s    c                   C   s   t  d¡ dS )zDisables ONNX logging.FNr;   r8   r8   r8   r9   r3   —   s    )KZ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   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-   r.   Z_internal.onnxruntimer/   r0   Z_OrtBackendr1   Z_OrtBackendOptionsr2   Z_OrtExecutionProviderÚ__all__Ú
__module__Zproducer_nameZPRODUCER_VERSIONZproducer_versionÚ
deprecatedr5   Z_jit_is_onnx_log_enabledZis_onnx_log_enabledr4   r3   Z_jit_set_onnx_log_output_streamZset_log_streamZ_jit_onnx_logÚlogr8   r8   r8   r9   Ú<module>   s˜   L,(	Ò2  ÿ
	