o
    êÜÓh‘  ã                   @   s`   d dl mZmZmZmZmZmZmZmZ d dl	m
Z
 d dlmZmZ d dlmZ d dl	mZ dS )é    )Úgrad_and_valueÚvjpÚjvpÚjacrevÚjacfwdÚhessianÚfunctionalizeÚ	linearize)Úgrad)Úfunctional_callÚstack_module_state)Úreplace_all_batch_norm_modules_)ÚvmapN)Ú!torch._functorch.eager_transformsr   r   r   r   r   r   r   r	   Útorch._functorch.apisr
   Ú torch._functorch.functional_callr   r   Ú'torch._functorch.batch_norm_replacementr   r   © r   r   úI/var/www/html/ai/venv/lib/python3.10/site-packages/torch/func/__init__.pyÚ<module>   s
   ( 
