Searched refs:runGradCallbackForVariable (Results 1 – 4 of 4) sorted by relevance
359 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::Reducer1701 context_ptr->runGradCallbackForVariable(variable, std::move(cb)); in runGradCallbackForVariable()
310 void runGradCallbackForVariable(at::Tensor& variable, GradCallback&& cb);
62 void runGradCallbackForVariable(
248 void DistAutogradContext::runGradCallbackForVariable( in runGradCallbackForVariable() function in torch::distributed::autograd::DistAutogradContext