U
    Ò9%e›  ã                   @   s0   d dl mZ d dlmZmZ daedd„ ƒZdS )é    )Úcontextmanager)Ú_vmap_add_layersÚ_vmap_remove_layersFc                 c   sL   t rt‚tdd„ | D ƒƒ}t|ƒ}zt|ƒ da d V  W 5 da t|ƒ X d S )Nc                 s   s$   | ]}t |tƒs|j|jfV  qd S )N)Ú
isinstanceÚintÚ_levelÚsize)Ú.0Úd© r   úY/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/functorch/dim/batch_tensor.pyÚ	<genexpr>   s     
 z!_enable_layers.<locals>.<genexpr>FT)Ú_enabledÚAssertionErrorÚsortedÚlenr   r   )ZdimsÚinputÚnr   r   r   Ú_enable_layers   s    
r   N)Ú
contextlibr   Ztorch._C._functorchr   r   r   r   r   r   r   r   Ú<module>   s   