Home
last modified time | relevance | path

Searched refs:release_variables (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/tools/autograd/
H A Dgen_autograd_functions.py544 release_variables: list[str] = []
579 release_variables.append(f"{name}_.reset_data();")
616 release_variables.append(f"{name}_.clear();")
617 release_variables.append(f"{name}_released_ = true;")
639 release_variables.append(f"{name}_.clear();")
640 release_variables.append(f"{name}_released_ = true;")
739 release_variables.append(f"{name}.clear();")
813 if len(release_variables) > 0:
913 release_variables=release_variables,
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dfunction.cpp59 func->release_variables(); in gatherFunctions()
95 function->release_variables(); in deleteNode()
H A Dcustom_function.h186 void release_variables() override;
468 void CppNode<T>::release_variables() { in release_variables() function
H A Dpython_function.h42 void release_variables() override;
H A Dfunction.h542 virtual void release_variables() {} in release_variables() function
H A Dpython_function.cpp269 auto PyNode::release_variables() -> void { in release_variables() function in torch::autograd::PyNode
1209 cdata->release_variables(); in THPFunction_maybe_clear_saved_tensors()
H A Dengine.cpp1043 fn.release_variables(); in evaluate_function()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/functions/
H A Dtensor.h170 void release_variables() override;
H A Dtensor.cpp175 void CopySlices::release_variables() { in release_variables() function in torch::autograd::CopySlices
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dgraph_executor.cpp180 void release_variables() { in release_variables() function
376 void release_variables() override { in release_variables() function
377 captures_.release_variables(); in release_variables()
/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dpython_compiled_autograd.cpp676 call->node->release_variables(); in _compiled_autograd_impl()