Home
last modified time | relevance | path

Searched refs:runEngineAndAccumulateGradients (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/distributed/autograd/engine/
Ddist_engine.cpp403 runEngineAndAccumulateGradients( in runEngineAndAccumulateGradients() function in torch::distributed::autograd::DistEngine
494 auto accumulateGradFuture = runEngineAndAccumulateGradients( in executeSendFunctionAsync()
597 runEngineAndAccumulateGradients(autogradContext, graphRoot, outputEdges) in execute()
Ddist_engine.h128 c10::intrusive_ptr<c10::ivalue::Future> runEngineAndAccumulateGradients(