U
    dh                     @   s0   d dl Z d dlmZmZ ee jjdddZdS )    N)
OpOverloadOpOverloadPacketopgraphc                 C   s:   t | trtd|  t | ts&ttj| j| d S )Nz;Must pass specific op overload, not overload packet, found )
isinstancer   AssertionErrorr   torch_CZ&_jit_register_decomposition_for_schemaZ_schemar    r   B/tmp/pip-unpacked-wheel-ua33x9lu/torch/jit/_decomposition_utils.py_register_decomposition   s    r   )r	   Z
torch._opsr   r   r
   ZGraphr   r   r   r   r   <module>   s   