o
    Zh                     @   sd   d dl mZmZmZ d dlmZ d dlmZmZm	Z	m
Z
mZmZmZmZ d dlmZmZ g dZdS )    )gradgrad_and_valuevmap)replace_all_batch_norm_modules_)debug_unwrapfunctionalizehessianjacfwdjacrevjvp	linearizevjp)functional_callstack_module_state)r   r   r   r   r   r   r	   r
   r   r   r   r   r   r   N)Ztorch._functorch.apisr   r   r   Z'torch._functorch.batch_norm_replacementr   Z!torch._functorch.eager_transformsr   r   r   r	   r
   r   r   r   Z torch._functorch.functional_callr   r   __all__ r   r   B/var/www/auris/lib/python3.10/site-packages/torch/func/__init__.py<module>   s
    (
