a h@s0ddlZddlmZmZeejjdddZdS)N) OpOverloadOpOverloadPacketopgraphcCs:t|trJd|t|ts&Jtj|j|dS)Nz;Must pass specific op overload, not overload packet, found ) isinstancerrtorch_CZ&_jit_register_decomposition_for_schemaZ_schemarr L/var/www/auris/lib/python3.9/site-packages/torch/jit/_decomposition_utils.py_register_decompositions  r )rZ torch._opsrrr ZGraphr r r r r s