Home
last modified time | relevance | path

Searched refs:SwapSavedVariables (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/functions/
H A Dbasic_ops.h25 SwapSavedVariables& saved) override;
75 SwapSavedVariables& saved) override { in apply_with_saved()
98 SwapSavedVariables& saved) override;
H A Dtensor.h21 SwapSavedVariables& saved) override;
174 SwapSavedVariables& saved) override;
H A Dbasic_ops.cpp26 SwapSavedVariables& saved) { in apply_with_saved()
74 SwapSavedVariables& saved) { in apply_with_saved()
H A Dtensor.cpp46 SwapSavedVariables& saved) { in apply_with_saved()
192 SwapSavedVariables& saved) { in apply_with_saved()
H A Daccumulate_grad.cpp80 SwapSavedVariables& saved) { in apply_with_saved()
H A Daccumulate_grad.h268 SwapSavedVariables& saved) override;
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dfunction_hook.h10 class SwapSavedVariables; variable
57 torch::dynamo::autograd::SwapSavedVariables&) { in apply_with_saved()
H A Dpython_hook.h8 class SwapSavedVariables; variable
51 torch::dynamo::autograd::SwapSavedVariables& saved) override;
H A Dfunction.h39 using torch::dynamo::autograd::SwapSavedVariables;
585 SwapSavedVariables& saved) { in apply_with_saved()
H A Dpython_function.h49 SwapSavedVariables& saved) override;
H A Dpython_hook.cpp243 torch::dynamo::autograd::SwapSavedVariables& saved) { in apply_with_saved()
H A Dcustom_function.h222 SwapSavedVariables& saved) override { in apply_with_saved()
H A Dpython_function.cpp357 SwapSavedVariables& saved) { in apply_with_saved()
/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dcompiled_autograd.h581 class SwapSavedVariables {
771 SwapSavedVariables( in SwapSavedVariables() function
H A Dpython_compiled_autograd.cpp617 SwapSavedVariables saved(compiler_call, state, py_compiler.get(), call); in _compiled_autograd_impl()