קgbddlmZmZmZmZmZmZmZddlm 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_transformsrrrrrrr torch._functorch.apisr r torch._functorch.functional_callr r 'torch._functorch.batch_norm_replacementrrO/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/func/__init__.pyrs76666666PPPPPPPPSSSSSS&&&&&&&&r