o
    Zh                     @   s2   d dl Z d dlmZmZ dede jjfddZdS )    N)
OpOverloadOpOverloadPacketopgraphc                 C   s:   t | trJ d|  t | tsJ tj| j| d S )Nz;Must pass specific op overload, not overload packet, found )
isinstancer   r   torch_CZ&_jit_register_decomposition_for_schemaZ_schema)r   r    r	   M/var/www/auris/lib/python3.10/site-packages/torch/jit/_decomposition_utils.py_register_decomposition   s   r   )r   Z
torch._opsr   r   r   ZGraphr   r	   r	   r	   r
   <module>   s   