Searched refs:saved_variables (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/pytorch/tools/autograd/ |
H A D | gen_autograd_functions.py | 543 saved_variables: list[str] = [] 578 saved_variables.append(f"SavedVariable {name}_;") 612 saved_variables.append(f"std::vector<SavedVariable> {name}_;") 613 saved_variables.append(f"bool {name}_released_ = false;") 635 saved_variables.append(f"std::vector<SavedVariable> {name}_;") 636 saved_variables.append(f"bool {name}_released_ = false;") 656 saved_variables.append(f"std::vector<int64_t> {name};") 663 saved_variables.append(f"std::vector<c10::SymInt> {name};") 670 saved_variables.append(f"c10::OptionalArray<int64_t> {name};") 677 saved_variables.append(f"c10::OptionalArray<c10::SymInt> {name};") [all …]
|
H A D | load_derivatives.py | 208 formula, saved_inputs = saved_variables(formula, arguments, var_names) 209 formula, saved_outputs = saved_variables(formula, named_returns, var_names) 751 def saved_variables( function
|
H A D | gen_variable_type.py | 1424 saved_variables: Sequence[SavedAttribute], 1430 for arg in sorted(saved_variables, key=lambda sa: str(sa.nctype.name)):
|
/aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
H A D | python_function.cpp | 80 auto& saved_variables = self->saved_variables; in unpack_saved_variables() local 81 if (saved_variables.empty()) in unpack_saved_variables() 84 auto num_saved = saved_variables.size(); in unpack_saved_variables() 98 auto unpacked_var = saved_variables[i].unpack(saved_for); in unpack_saved_variables() 277 f->saved_variables.clear(); in release_variables() 333 args.collect(f->saved_variables, true); // always unpacked as output in eager in compiled_args() 361 saved.before(f->saved_variables); in apply_with_saved() 391 saved.after(f->saved_variables); in apply_with_saved() 497 self->saved_variables.clear(); in THPFunction_clear() 523 self->saved_variables.~vector(); in THPFunction_dealloc() [all …]
|
H A D | python_function.h | 133 std::vector<torch::autograd::SavedVariable> saved_variables; member
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_autograd.py | 3083 var1, var2 = ctx.saved_variables
|