U ʹj @sdZddlmZddlZddlmZddlmZddlm Z ddl m Z dd d d d gZ Gd d d e ZGdddeZGdd d eZGdd d eZGdd d eZdS)zONNX exporter exceptions.) annotationsN)Optional)_C) _constants) diagnosticsOnnxExporterErrorOnnxExporterWarning CheckerErrorUnsupportedOperatorErrorSymbolicValueErrorc@seZdZdZdS)rz1Base class for all warnings in the ONNX exporter.N__name__ __module__ __qualname____doc__rrP/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/onnx/errors.pyrsc@seZdZdZdS)rz#Errors raised by the ONNX exporter.Nr rrrrrsc@seZdZdZdS)r z2Raised when ONNX checker detects an invalid model.Nr rrrrr scs*eZdZdZddddfdd ZZS)r z7Raised when an operator is unsupported by the exporter.strintz Optional[int])nameversionsupported_versioncs|dk r2tjj}||||}t|tjj|nZ|drhtjj}|||t j }t|tjj|n$tjj }||}t|tjj|t |dS)N)zaten::zprim::z quantized::)rrulesZ)operator_supported_in_newer_opset_versionZformat_messageZdiagnoseZlevelsERROR startswithZ"missing_standard_symbolic_functionrZPYTORCH_GITHUB_ISSUES_URLZ missing_custom_symbolic_functionsuper__init__)selfrrrZdiagnostic_rulemsg __class__rrr)s"  z!UnsupportedOperatorError.__init__r rrrr __classcell__rrrrr &scs(eZdZdZdddfdd ZZS)r z+Errors around TorchScript values and nodes.rz_C.Value)rvaluec 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 Rsz.SymbolicValueError.__init__..z Emptyz Outputs: css.|]&\}}d|d|d|dVqdSr&r')r)r*outputrrrr+[sz zv Failed to obtain its input and output for debugging. Please refer to the TorchScript graph for debugging information.) r(nodekindZ sourceRangetextwrapindentjoin enumerateinputsoutputsAttributeErrorrr)rrr#messageZ code_locationrrrrAs<&      zSymbolicValueError.__init__r!rrrrr >s)r __future__rr/typingrZtorchrZ torch.onnxrZtorch.onnx._internalr__all__ UserWarningr RuntimeErrorrr r r rrrrs"