Home
last modified time | relevance | path

Searched refs:get_saved_variables (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A D_functions.cpp89 auto input = ctx->get_saved_variables()[0]; in backward()
90 auto output = ctx->get_saved_variables()[1]; in backward()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_autograd_functions.cpp37 auto saved = ctx->get_saved_variables(); in backward()
/aosp_15_r20/external/pytorch/test/custom_operator/
H A Dop.cpp45 auto saved = ctx->get_saved_variables(); in backward()
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dautograd.cpp311 auto saved = ctx->get_saved_variables(); in TEST()
346 auto saved = ctx->get_saved_variables(); in TEST()
405 auto saved = ctx->get_saved_variables(); in TEST()
745 auto saved = ctx->get_saved_variables(); in TEST()
H A Dinference_mode.cpp626 auto saved = ctx->get_saved_variables(); in TEST()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dcustom_function.cpp528 variable_list AutogradContext::get_saved_variables() const { in get_saved_variables() function in torch::autograd::AutogradContext
H A Dcustom_function.h147 variable_list get_saved_variables() const;