o
    h                     @   s   d dl Z d dlmZ d dlmZ d dlmZ d dlm	Z	 deege	f deege	f fdd	Z
deeege	f deeege	f fd
dZdS )    N)Callable)'NativeFunctionWithDifferentiabilityInfo)native_function_manager)Tfuncreturnc                    s$   t  dtdtf fdd}|S )Nfr   c                    s6   t | j  | W  d    S 1 sw   Y  d S Nr   r   )r   r    v/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torchgen/packaged/autograd/context.pywrapper   s   $zAwith_native_function_with_differentiability_info.<locals>.wrapper)	functoolswrapsNFWDIr   r   r   r   r   r   0with_native_function_with_differentiability_info   s   r   c                    s(   t  dtdtdtf fdd}|S )Nr   keyr   c                    s8   t | j  | |W  d    S 1 sw   Y  d S r	   r
   )r   r   r   r   r   r      s   $zIwith_native_function_with_differentiability_info_and_key.<locals>.wrapper)r   r   r   strr   r   r   r   r   8with_native_function_with_differentiability_info_and_key   s   r   )r   typingr   torchgen.api.autogradr   r   torchgen.contextr   torchgen.utilsr   r   r   r   r   r   r   r   <module>   s    
