o
    ÜÜÓhŠ  ã                   @   sD   d dl mZ d dlZG dd„ dƒZdeejjejjf fdd„ZdS )é    )ÚUnionNc                   @   sB   e Zd Zdejjdeejjejjf fdd„Z	dd„ Z
dd„ Zd	S )
Ú_InsertPointÚinsert_point_graphÚinsert_pointc                 C   s   || _ || _d | _d S ©N)r   ÚgÚguard)Úselfr   r   © r
   úI/var/www/html/ai/venv/lib/python3.10/site-packages/torch/jit/_ir_utils.pyÚ__init__   s   
z_InsertPoint.__init__c                 C   s   | j  ¡ | _| j  | j¡ d S r   )r   ÚinsertPointÚprev_insert_pointÚsetInsertPointr   )r	   r
   r
   r   Ú	__enter__   s   z_InsertPoint.__enter__c                 G   s   | j  | j¡ d S r   )r   r   r   )r	   Úargsr
   r
   r   Ú__exit__   s   z_InsertPoint.__exit__N)Ú__name__Ú
__module__Ú__qualname__ÚtorchÚ_CÚGraphr   ÚNodeÚBlockr   r   r   r
   r
   r
   r   r      s    þ
ý	r   r   c                 C   s
   t | |ƒS r   )r   )r	   r   r
   r
   r   Úinsert_point_guard   s   
r   )Útypingr   r   r   r   r   r   r   r
   r
   r
   r   Ú<module>   s    "