o
    fi                     @   s6   d dl Z d dlmZmZ dede jjddfddZdS )    N)
OpOverloadOpOverloadPacketopgraphreturnc                 C   sH   t | trtd|  t | tstdt|  tj| j| d S )Nz;Must pass specific op overload, not overload packet, found zExpected OpOverload, got )	
isinstancer   AssertionErrorr   typetorch_C&_jit_register_decomposition_for_schema_schema)r   r    r   h/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torch/jit/_decomposition_utils.py_register_decomposition   s   

r   )r
   
torch._opsr   r   r   Graphr   r   r   r   r   <module>   s   