Home
last modified time | relevance | path

Searched refs:save_variables (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/tools/autograd/
H A Dgen_variable_type.py1276 save_input_stmts = save_variables(info.all_saved_inputs, False, guard_for)
1284 setup.extend(save_variables(info.all_saved_inputs, False, guard_for))
1423 def save_variables( function
1767 stmts = save_variables(info.all_saved_outputs, True)
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dcustom_function.h169 void save_variables();
477 ctx_.save_variables(); in save_variables_to_ctx()
H A Dcustom_function.cpp512 void AutogradContext::save_variables() { in save_variables() function in torch::autograd::AutogradContext