o ߗi@s`ddlmZmZmZmZmZmZmZddlm Z m Z ddl m Z m Z ddlmZddlmZdS))vjpjvpjacrevjacfwdhessian functionalize linearize)gradgrad_and_value)functional_callstack_module_state)replace_all_batch_norm_modules_)vmapN)!torch._functorch.eager_transformsrrrrrrrtorch._functorch.apisr r torch._functorch.functional_callr r 'torch._functorch.batch_norm_replacementr rrrQ/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/torch/func/__init__.pys $