&VjifddlmZmZmZddlmZddlmZmZm Z m Z m Z m Z m Z mZddlmZmZgdZdS))gradgrad_and_valuevmap)replace_all_batch_norm_modules_) debug_unwrap functionalizehessianjacfwdjacrevjvp linearizevjp)functional_callstack_module_state)rrrrrr r r r r rrrrN)torch._functorch.apisrrr'torch._functorch.batch_norm_replacementr!torch._functorch.eager_transformsrrr r r r r r torch._functorch.functional_callrr__all__M/root/voice-cloning/.venv/lib/python3.11/site-packages/torch/func/__init__.pyrs<<<<<<<<<<SSSSSS                    QPPPPPPP   r