U d@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{}) itertoolschainrZonnx_stable_opsetsonnx_main_opset importlib import_moduleformatr )Z opset_versionmoduler@/tmp/pip-unpacked-wheel-ua33x9lu/torch/onnx/symbolic_registry.py_import_symbolic_opsetssrdomainversioncCs$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)rrrZop_nameZop_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_lenlenZ_listlistZ_anyanyZ_allall) r rinspect getmembers isinstancetypehasattr isfunctionappendr)rmembersZdomain_opname_opsobjopsoprrrrQs("        rcCs ||ftkSr)r rrrrrisrcCsD|dks|dkrtdt||s0it||f<|t||f|<dS)NzHONNX export failed. The ONNX domain and/or version to register are None.)warningswarnrr )opnamer5rrrrrr ns  r r8rrcCs6|dks|dkrtd||ftko4|t||fkSNz|}|tjkr:ddt|D}||f|kr0|S|d7}qdS)NcSsg|]}|d|dfqS)rr#r).0r5rrr sz,get_op_supported_version..r#)rr r)r8rrrr4rrrget_op_supported_versions   r>)r8rrreturncCsB|dks|dkrtdt|||s2t|||t||f|Sr:)r6r7rUnsupportedOperatorErrorr r9rrrget_registered_ops    rAcs&eZdZeeedfdd ZZS)r@)rr8rcsrt|||}|dkrPd|d|d|d}|dk rF|d|d7}qb|d7}nd |d|d }t|dS) N>r*aten quantizedprimzExporting 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.)r>super__init__)selfrr8rZsupported_versionmsg __class__rrrFs   z!UnsupportedOperatorError.__init__)__name__ __module__ __qualname__strintrF __classcell__rrrIrr@sr@)"rr+r r6typingrrrrrZtorchrZ torch.onnxrValueZ_SymbolicFunctionr rNrO__annotations__r rrr!rrrr rr;r>rA RuntimeErrorr@rrrrs6