o
    HÆ&iŸ  ã                   @   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)Z!torch._functorch.eager_transformsr   r   r   r   r   r   r   r	   Ztorch._functorch.apisr
   Z torch._functorch.functional_callr   r   Z'torch._functorch.batch_norm_replacementr   r   © r   r   ú>C:\wamp64\www\opt\env\Lib\site-packages\torch/func/__init__.pyÚ<module>   s
   ( 
