Home
last modified time | relevance | path

Searched refs:saved_variables (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/tools/autograd/
H A Dgen_autograd_functions.py543 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 Dload_derivatives.py208 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 Dgen_variable_type.py1424 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 Dpython_function.cpp80 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 Dpython_function.h133 std::vector<torch::autograd::SavedVariable> saved_variables; member
/aosp_15_r20/external/pytorch/test/
H A Dtest_autograd.py3083 var1, var2 = ctx.saved_variables