
    Χg                         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Z
deeege	f         deeege	f         fd	ZdS )
    N)Callable)'NativeFunctionWithDifferentiabilityInfo)native_function_manager)Tfuncreturnc                 d     t          j                   dt          dt          f fd            }|S )Nfr   c                 v    t          | j                  5   |           cd d d            S # 1 swxY w Y   d S Nr   r   )r
   r   s    ^/var/www/html/ai-engine/env/lib/python3.11/site-packages/torchgen/packaged/autograd/context.pywrapperzAwith_native_function_with_differentiability_info.<locals>.wrapper   s    $QV,, 	 	477	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s   .22)	functoolswrapsNFWDIr   r   r   s   ` r   0with_native_function_with_differentiability_infor      sM     _T5 Q       N    c                 r     t          j                   dt          dt          dt          f fd            }|S )Nr
   keyr   c                 x    t          | j                  5   | |          cd d d            S # 1 swxY w Y   d S r   r   )r
   r   r   s     r   r   zIwith_native_function_with_differentiability_info_and_key.<locals>.wrapper   s    $QV,, 	  	 43<<	  	  	  	  	  	  	  	  	  	  	  	  	  	  	  	  	  	 s   /33)r   r   r   strr   r   s   ` r   8with_native_function_with_differentiability_info_and_keyr      sT     _T 5  s  q             Nr   )r   typingr   torchgen.api.autogradr   r   torchgen.contextr   torchgen.utilsr   r   r   r    r   r   <module>r       s              R R R R R R 4 4 4 4 4 4      

E7A:
ugqj   
E3<?
#uclAo     r   