Searched refs:retrieveGraphTask (Results 1 – 4 of 4) sorted by relevance
412 auto graphTask = autogradContext->retrieveGraphTask(); in runEngineAndAccumulateGradients()492 auto graphTask = autogradContext->retrieveGraphTask(); in executeSendFunctionAsync()548 auto graphTask = autogradContext->retrieveGraphTask(); in executeSendFunctionAsync()614 autogradContext->retrieveGraphTask()->future_result_; in cleanupBackwardPass()
49 sharedContext->retrieveGraphTask()->keep_graph_); in apply()
94 std::shared_ptr<torch::autograd::GraphTask> retrieveGraphTask();
115 retrieveGraphTask() { in retrieveGraphTask() function in torch::distributed::autograd::DistAutogradContext