
    sg                         d dl mZ d dlZ G d d      Zdeej                  j
                  ej                  j                  f   fdZy)    )UnionNc                       e Zd Zdej                  j
                  deej                  j                  ej                  j                  f   fdZ	d Z
d Zy)_InsertPointinsert_point_graphinsert_pointc                 .    || _         || _        d | _        y N)r   gguard)selfr   r   s      F/var/www/html/venv/lib/python3.12/site-packages/torch/jit/_ir_utils.py__init__z_InsertPoint.__init__   s    
 )#
    c                     | j                   j                         | _        | j                   j                  | j                         y r	   )r
   insertPointprev_insert_pointsetInsertPointr   )r   s    r   	__enter__z_InsertPoint.__enter__   s/    !%!3!3!5d//0r   c                 N    | j                   j                  | j                         y r	   )r
   r   r   )r   argss     r   __exit__z_InsertPoint.__exit__   s    d445r   N)__name__
__module____qualname__torch_CGraphr   NodeBlockr   r   r    r   r   r   r      sE    !HHNN EHHMM588>>9:16r   r   r   c                     t        | |      S r	   )r   )r   r   s     r   insert_point_guardr"      s    l++r   )typingr   r   r   r   r   r   r"   r    r   r   <module>r$      s;     6 6$,51N+O ,r   