U
    yh                     @   s`   d dl mZmZmZmZmZmZmZ d dlm	Z	m
Z
 d dlmZmZ d dlmZ d dlmZ dS )    )vjpjvpjacrevjacfwdhessianfunctionalize	linearize)gradgrad_and_value)functional_callstack_module_state)replace_all_batch_norm_modules_)vmapN)Z!torch._functorch.eager_transformsr   r   r   r   r   r   r   Ztorch._functorch.apisr	   r
   Z torch._functorch.functional_callr   r   Z'torch._functorch.batch_norm_replacementr   r    r   r   E/var/www/html/venv/lib/python3.8/site-packages/torch/func/__init__.py<module>   s   $	