o wZh@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__rr@/var/www/auris/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 fdd ZZS) rz7Raised when an operator is unsupported by the exporter.namestrversionintsupported_version int | Nonecsddlm}ddlm}|dur%|jj}||||}|||jj |n-| dr@|jj }||||j }|||jj |n|jj }||}|||jj |t|dS)Nr) _constants) diagnostics)zaten::zprim::z quantized::)Z torch.onnxrZtorch.onnx._internalrrulesZ)operator_supported_in_newer_opset_versionZformat_messageZdiagnoseZlevelsERROR startswithZ"missing_standard_symbolic_functionZPYTORCH_GITHUB_ISSUES_URLZ missing_custom_symbolic_functionsuper__init__)selfrrrrrZdiagnostic_rulemsg __class__rrr!s"    z!UnsupportedOperatorError.__init__)rrrrrrr r r r r __classcell__rrr!rrrcs"eZdZdZdfdd ZZS) rz+Errors around TorchScript values and nodes.r rvalue_C.Valuec 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).0iZinput_rrr O  z.SymbolicValueError.__init__..z Emptyz Outputs: csr*r+r,)r.r/outputrrrr0Xr1z zv Failed to obtain its input and output for debugging. Please refer to the TorchScript graph for debugging information.) r-nodekindZ sourceRangetextwrapindentjoin enumerateZinputsZoutputsAttributeErrorrr)rr r&messageZ code_locationr!rrr>sB      zSymbolicValueError.__init__)r rr&r'r#rrr!rr9r%r)r __future__r__all__r5typingrZtorchr UserWarningr RuntimeErrorrrrrrrrs