a 8Sic@spUddlZddlZddlZddlZddlmZmZmZmZm Z ddl m Z ddl m Z ede e jee jffZiaeeeefeeeffed<iaee eefefed<dd Zeed d d Zeeed ddZeed ddZedddZeed ddZddZeeedddZeeedddZeeedddZeeeed d!d"ZGd#d$d$e Z!dS)%N)AnyCallableDictTupleUnion)_C) _constants. _registry_symbolic_versionscCs4ttjtjgD]}td|}|t|<qdS)Nztorch.onnx.symbolic_opset{}) itertoolschainronnx_stable_opsetsonnx_main_opset importlib import_moduleformatr ) opset_versionmodulerX/var/www/html/django/DPS/env/lib/python3.9/site-packages/torch/onnx/symbolic_registry.py_import_symbolic_opsetss rdomainversioncCs$t||sit||f<t||dSN)is_registered_versionr register_ops_in_versionrrrrregister_version's  rrr iter_versioncCs2t|D]$\}}}t|||st||||qdSr)get_ops_in_versionis_registered_op register_op)rrrop_nameop_funcrrrregister_ops_helper.s r%cCsD|}|dkr4t||||dkr*|d}q|d}qt||ddS)N )r%rrrrr4s   r)rcCs ts ttt|}g}|D]}t|dtr~t|ddr~tj|dtjd}|D]"}||dj |d|dfqXq t|dr |ddkrd|df}|ddkrd|df}|dd krd |df}|dd krd |df}|d |d|dfq |S)Nr'r) predicater_lenlen_listlist_anyany_allall) r rinspect getmembers isinstancetypehasattr isfunctionappendr)rmembersZdomain_opname_opsobjopsoprrrr Qs("        r cCs ||ftvSr)r rrrrrisrcCsD|dus|durtdt||s0it||f<|t||f|<dS)NzHONNX export failed. The ONNX domain and/or version to register are None.)warningswarnrr )opnamer<rrrrrr"ns  r"r?rrcCs6|dus|durtd||ftvo4|t||fvSNzr r@rrrr!ys r!cCsHt|||r2t||f|=t||fsDt||f=ntd|ddS)Nz The opname z is not registered.)r!r r=r>r@rrr unregister_ops    rBcCs>|}|tjkr:ddt|D}||f|vr0|S|d7}qdS)NcSsg|]}|d|dfqS)rr'r).0r<rrr z,get_op_supported_version..r')rrr )r?rrrr;rrrget_op_supported_versions   rF)r?rrreturncCsB|dus|durtdt|||s2t|||t||f|SrA)r=r>r!UnsupportedOperatorErrorr r@rrrget_registered_ops    rIcs&eZdZeeedfdd ZZS)rH)rr?rcsrt|||}|dvrPd|d|d|d}|durF|d|d7}qb|d7}nd |d|d }t|dS) N>r1primaten 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.zOPlease feel free to request support or submit a pull request on PyTorch GitHub.z>ONNX export failed on an operator with unrecognized namespace zo. If you are trying to export a custom operator, make sure you registered it with the right domain and version.)rFsuper__init__)selfrr?rZsupported_versionmsg __class__rrrNs   z!UnsupportedOperatorError.__init__)__name__ __module__ __qualname__strintrN __classcell__rrrQrrHsrH)"rr2r r=typingrrrrrtorchr torch.onnxrValueZ_SymbolicFunctionr rVrW__annotations__r rrr%rr rr"r!rBrFrI RuntimeErrorrHrrrrs4