o i @sdZddlmZgdZddlZddlmZerddlmZGddde Z Gd d d e Z Gd d d e Z Gd dde ZdS)zONNX exporter exceptions.) annotations)OnnxExporterWarningSymbolicValueErrorUnsupportedOperatorErrorN) TYPE_CHECKING)_Cc@eZdZdZdS)rzWarnings in the ONNX exporter.N__name__ __module__ __qualname____doc__rrS/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/torch/onnx/errors.pyrrc@r)OnnxExporterErrorzSErrors raised by the ONNX exporter. This is the base class for all exporter errors.Nr rrrrrrrcs"eZdZdZd fd d ZZS) rz7Raised when an operator is unsupported by the exporter.namestrversionintsupported_version int | NonereturnNonecs\|durd|d|d|d}n|dr d|d|d}nd|d}t|dS) NzExporting the operator 'z' to ONNX opset version zB is not supported. Support for this operator was added in version z!, try exporting with this version)zaten::zprim::z quantized::z is not supportedz>ONNX export failed on an operator with unrecognized namespace zo. If you are trying to export a custom operator, make sure you registered it with the right domain and version.) startswithsuper__init__)selfrrrmsg __class__rrr!s  z!UnsupportedOperatorError.__init__)rrrrrrrrr r r r r __classcell__rrrrrrcs"eZdZdZd fdd ZZS) rz+Errors around TorchScript values and nodes.rrvalue_C.Valuerrc s|d|d|d|d}|}|r$|d|d7}z6|d7}|tdd d d t|Dp>d d d d dd t| DpTd d7}Wn t yg|d7}Ynwt |dS)Nz [Caused by the value 'z ' (type 'z;') in the TorchScript graph. The containing node has kind 'z'.] z (node defined in )z zInputs:  cs0|]\}}d|d|d|dVqdSz #z: z (type 'z')Ntype).0iinput_rrr M  z.SymbolicValueError.__init__..z Emptyz Outputs: csr(r)r*)r,r-outputrrrr/Vr0z zv Failed to obtain its input and output for debugging. Please refer to the TorchScript graph for debugging information.) r+nodekind sourceRangetextwrapindentjoin enumerateinputsoutputsAttributeErrorrr)rrr$message code_locationrrrr<sB      zSymbolicValueError.__init__)rrr$r%rrr!rrrrr7r#r)r __future__r__all__r5typingrtorchr UserWarningr RuntimeErrorrrrrrrrs