
    [Th                     4    S SK Jr  S SKJrJr  Sq\S 5       rg)    )contextmanager)_vmap_add_layers_vmap_remove_layersFc              #      #    [         (       a   e[        S U  5       5      n[        U5      n [        U5        Sq S v   Sq [	        U5        g ! Sq [	        U5        f = f7f)Nc              3      #    U  H4  n[        U[        5      (       a  M  UR                  UR                  4v   M6     g 7f)N)
isinstanceint_levelsize).0ds     R/var/www/auris/envauris/lib/python3.13/site-packages/functorch/dim/batch_tensor.py	<genexpr>!_enable_layers.<locals>.<genexpr>   s)     Nt!:a;M%AHHaff%ts   >>TF)_enabledsortedlenr   r   )dimsinputns      r   _enable_layersr      s\      x<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   