o F h@s2ddlZddlmZmZdedejjfddZdS)N) OpOverloadOpOverloadPacketopgraphcCs:t|tr Jd|t|tsJtj|j|dS)Nz;Must pass specific op overload, not overload packet, found ) isinstancerrtorch_C&_jit_register_decomposition_for_schema_schema)rrr R/var/www/vscode/kcb/lib/python3.10/site-packages/torch/jit/_decomposition_utils.py_register_decompositionsr )r torch._opsrrrGraphr r r r r s