Home
last modified time | relevance | path

Searched refs:runGradCallbackForVariable (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Dreducer.cpp359 runGradCallbackForVariable(variable, [&](auto& grad) { in mark_variable_ready_dense()
442 runGradCallbackForVariable(variable, [&](auto& grad) { in mark_variable_ready_sparse()
661 runGradCallbackForVariable(variable, [&](auto& grad) { in autograd_hook()
1269 runGradCallbackForVariable(v, [&](auto& grad) { in initialize_bucket_views()
1464 runGradCallbackForVariable(variable, [&](auto& grad) { in copy_bucket_to_grad()
1551 runGradCallbackForVariable(variable, [&](auto& grad) { return true; }); in finalize_bucket_dense()
1568 runGradCallbackForVariable(variable, [&](auto& grad) { in finalize_bucket_dense()
1690 void Reducer::runGradCallbackForVariable( in runGradCallbackForVariable() function in c10d::Reducer
1701 context_ptr->runGradCallbackForVariable(variable, std::move(cb)); in runGradCallbackForVariable()
H A Dreducer.hpp310 void runGradCallbackForVariable(at::Tensor& variable, GradCallback&& cb);
/aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/context/
H A Dcontext.h62 void runGradCallbackForVariable(
H A Dcontext.cpp248 void DistAutogradContext::runGradCallbackForVariable( in runGradCallbackForVariable() function in torch::distributed::autograd::DistAutogradContext