U <cW@sdZddlmZddlZddlmZddlmZddlm Z ddl m Z dd d d d d gZ Gdd d e ZGdd d eZGdddeZGdd d eZGdd d eZGdd d eZdS)zONNX exporter exceptions.) annotationsN)Optional)_C) _constants) diagnosticsOnnxExporterErrorOnnxExporterWarningCallHintViolationWarning CheckerErrorUnsupportedOperatorErrorSymbolicValueErrorc@seZdZdZdS)rz1Base class for all warnings in the ONNX exporter.N__name__ __module__ __qualname____doc__rr5/tmp/pip-unpacked-wheel-gikjz4vx/torch/onnx/errors.pyrsc@seZdZdZdS)r zCWarning raised when a type hint is violated during a function call.Nr rrrrr sc@seZdZdZdS)rz#Errors raised by the ONNX exporter.Nr rrrrr!sc@seZdZdZdS)r z2Raised when ONNX checker detects an invalid model.Nr rrrrr 'scs,eZdZdZdddddfdd ZZS)r z7Raised when an operator is unsupported by the exporter.strintz Optional[int])domainop_nameversionsupported_versioncs|dkrd|d|d|d}|dk rd|d|d7}tjjtjjtjj|d|||fdq|d 7}|tj7}tjjtjj tjj|d||tjfdn8d |d|d }tjjtjj tjj|d|fdt |dS) N>atenprim quantizedzExporting the operator 'z::z' to ONNX opset version z is not supported. z/Support for this operator was added in version z", try exporting with this version.)Z message_argszPPlease feel free to request support or submit a pull request on PyTorch GitHub: z?ONNX export failed on an operator with unrecognized namespace 'zp'. If you are trying to export a custom operator, make sure you registered it with the right domain and version.) rcontextZdiagnoserulesZ)operator_supported_in_newer_opset_versionlevelsERRORrZPYTORCH_GITHUB_ISSUES_URLZ"missing_standard_symbolic_functionZ missing_custom_symbolic_functionsuper__init__)selfrrrrmsg __class__rrr#0s@      z!UnsupportedOperatorError.__init__rrrrr# __classcell__rrr&rr -scs(eZdZdZdddfdd ZZS)r z+Errors around TorchScript values and nodes.rz_C.Value)r%valuec s|d|d|d|d}|}|rH|d|d7}zl|d7}|tdd d d t|Dp|d d d d dd t| Dpd d7}Wnt k r|d7}YnXt |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:  css.|]&\}}d|d|d|dVqdSz #z: z (type 'z')Ntype).0iZinput_rrr usz.SymbolicValueError.__init__..z Emptyz Outputs: css.|]&\}}d|d|d|dVqdSr-r.)r0r1outputrrrr2~sz zv Failed to obtain its input and output for debugging. Please refer to the TorchScript graph for debugging information.) r/nodekindZ sourceRangetextwrapindentjoin enumerateinputsoutputsAttributeErrorr"r#)r$r%r*messageZ code_locationr&rrr#ds<&      zSymbolicValueError.__init__r(rrr&rr as)r __future__rr6typingrZtorchrZ torch.onnxrZtorch.onnx._internalr__all__ UserWarningrr RuntimeErrorrr r r rrrrs&      4