'}h dZddlmZddlZddlmZddlmZddlm Z ddl m Z gdZ Gd d e ZGd d eZGd deZGddeZGddeZy)zONNX exporter exceptions.) annotationsN)Optional)_C) _constants) diagnostics)OnnxExporterErrorOnnxExporterWarning CheckerErrorSymbolicValueErrorUnsupportedOperatorErrorceZdZdZy)r z1Base class for all warnings in the ONNX exporter.N__name__ __module__ __qualname____doc__P/var/www/html/test/engine/venv/lib/python3.12/site-packages/torch/onnx/errors.pyr r s;rr ceZdZdZy)rz#Errors raised by the ONNX exporter.Nrrrrrrs-rrceZdZdZy)r z2Raised when ONNX checker detects an invalid model.Nrrrrr r s<rr c$eZdZdZdfd ZxZS)r z7Raised when an operator is unsupported by the exporter.c|]tjj}|j|||}tj|tj j |n|jdrktjj}|j||tj}tj|tj j |nZtjj}|j|}tj|tj j |t|5|y)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__)selfnameversionsupported_versiondiagnostic_rulemsg __class__s rr%z!UnsupportedOperatorError.__init__)s  (!!KK "00w@QRC  +2D2D2J2JC PBC"-"3"3"V"V%44':#G#G$$_k6H6H6N6NPST"-"3"3"T"T%44T:$$_k6H6H6N6NPST r)r'strr(intr)z Optional[int]rrrrr% __classcell__r,s@rr r &sArr c$eZdZdZdfd ZxZS)r z+Errors around TorchScript values and nodes.c `|d|d|jd|jjd}|jj}|r |d|dz } |dz }|t j dd j d t|jjDxsd zd zd zd j d t|jjDxsd zdz }t|1|y#t$r|dz }Y wxYw)Nz [Caused by the value 'z ' (type 'z;') in the TorchScript graph. The containing node has kind 'z'.] z (node defined in )z zInputs:  c3VK|]!\}}d|d|d|jd#ywz #z: z (type 'z')Ntype).0iinput_s r z.SymbolicValueError.__init__..R7" )6$A3b &++-K"')z Emptyz Outputs: c3VK|]!\}}d|d|d|jd#ywr7r8)r:r;outputs rr=z.SymbolicValueError.__init__..[r>r?z zv Failed to obtain its input and output for debugging. Please refer to the TorchScript graph for debugging information.) r9nodekind sourceRangetextwrapindentjoin enumerateinputsoutputsAttributeErrorr$r%)r&r+valuemessage code_locationr,s rr%zSymbolicValueError.__init__As\e+E7)EJJL>J@@E @Q@Q@S?TTX Z   002  0qA AG  v G x "-6uzz|7J7J7L-M"'' # # "-6uzz|7K7K7M-N"''$) G8 !   S G s$B(DD-,D-)r+r-rLz_C.Valuer/r1s@rr r >s5)")"rr )r __future__rrEtypingrtorchr torch.onnxrtorch.onnx._internalr__all__ UserWarningr RuntimeErrorrr r r rrrrWs`"!,  +    $ 00,"*,"r