
    9j                     X    d dl Z d dlmZmZ dede j                  j
                  ddfdZy)    N)
OpOverloadOpOverloadPacketopgraphreturnc                     t        | t              rt        d|        t        | t              st        dt	        |              t
        j                  j                  | j                  |       y )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   s     ^/media/conek/DATA/Code/OCR/venv/lib/python3.12/site-packages/torch/jit/_decomposition_utils.py_register_decompositionr      s_    "&'I"N
 	
 b*%8b
CDD	HH33BJJF    )r   
torch._opsr   r   r   Graphr    r   r   <module>r      s1     3G
 G588>> Gd Gr   