
    Ph                         d dl Z d dlZd dlmZmZmZ d dlmZmZ e j                  d        Z
d Zd Zeeeedf   f   Zy)    N)*set_single_level_autograd_function_allowed*get_single_level_autograd_function_allowedunwrap_if_dead)UnionTuplec               #   v   K   	 t               } t        d       d  t        |        y # t                w xY ww)NT)r   r   )
prev_states    aC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/_functorch/utils.py%enable_single_level_autograd_functionr   
   s/     ??A
2482:>2:>s   9) 969c                 *    t        d | D              }|S )Nc              3   j   K   | ]+  }t        |t        j                        rt        |      n| - y wN)
isinstancetorchTensorr   ).0args     r
   	<genexpr>z'unwrap_dead_wrappers.<locals>.<genexpr>   s/      C  *#u||<s#Es   13)tuple)argsresults     r
   unwrap_dead_wrappersr      s       F M    c                       fd}|S )Nc                     | _         | S r   )
__module__)fnmodules    r
   wrapperzexposed_in.<locals>.wrapper$   s    	r    )r   r   s   ` r
   
exposed_inr!   #   s     Nr   .)
contextlibr   torch._C._functorchr   r   r   typingr   r   contextmanagerr   r   r!   int	argnums_tr    r   r
   <module>r(      sU      
  ? ?  #uS#X&'	r   