
    Ph                     4    d dl mZ d dlmZmZ daed        Zy)    )contextmanager)_vmap_add_layers_vmap_remove_layersFc              #      K   t         rJ t        d | D              }t        |      }	 t        |       da d  da t	        |       y # da t	        |       w xY ww)Nc              3   n   K   | ]-  }t        |t              r|j                  |j                  f / y w)N)
isinstanceint_levelsize).0ds     eC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\functorch/dim/batch_tensor.py	<genexpr>z!_enable_layers.<locals>.<genexpr>   s'     Nt!:a;MAHHaff%ts   55TF)_enabledsortedlenr   r   )dimsinputns      r   _enable_layersr      s[      <NtNNEE
AA As   &AA AAAN)
contextlibr   torch._C._functorchr   r   r   r        r   <module>r      s&    & E  r   