o h @sdZddlmZddlZddlmZddlmZddlm Z ddl m Z gdZ Gd d d e ZGd d d eZGd ddeZGdddeZGdddeZdS)zONNX exporter exceptions.) annotationsN)Optional)_C) _constants) diagnostics)OnnxExporterErrorOnnxExporterWarning CheckerErrorUnsupportedOperatorErrorSymbolicValueErrorc@eZdZdZdS)rz1Base class for all warnings in the ONNX exporter.N__name__ __module__ __qualname____doc__rrG/var/www/html/ai/venv/lib/python3.10/site-packages/torch/onnx/errors.pyrrc@r )rz#Errors raised by the ONNX exporter.Nr rrrrrrrc@r )r z2Raised when ONNX checker detects an invalid model.Nr rrrrr rr cs"eZdZdZd fdd ZZS) r z7Raised when an operator is unsupported by the exporter.namestrversionintsupported_version Optional[int]cs|durtjj}||||}t|tjj|n-|dr4tjj}|||t j }t|tjj|ntjj }||}t|tjj|t |dS)N)zaten::zprim::z quantized::)rrules)operator_supported_in_newer_opset_versionformat_messagediagnoselevelsERROR startswith"missing_standard_symbolic_functionrPYTORCH_GITHUB_ISSUES_URL missing_custom_symbolic_functionsuper__init__)selfrrrdiagnostic_rulemsg __class__rrr&)s  z!UnsupportedOperatorError.__init__)rrrrrrrrrrr& __classcell__rrr*rr &r cs"eZdZdZdfdd ZZS) r z+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).0iinput_rrr R  z.SymbolicValueError.__init__..z Emptyz Outputs: csr3r4r5)r7r8outputrrrr:[r;z zv Failed to obtain its input and output for debugging. Please refer to the TorchScript graph for debugging information.) r6nodekind sourceRangetextwrapindentjoin enumerateinputsoutputsAttributeErrorr%r&)r'r)r/message code_locationr*rrr&AsB      zSymbolicValueError.__init__)r)rr/r0r,rrr*rr >r.r )r __future__rr@typingrtorchr torch.onnxrtorch.onnx._internalr__all__ UserWarningr RuntimeErrorrr r r rrrrs