Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/distributed/autograd/engine/
Ddist_engine.cpp465 const auto& send_backward_stream = sendFunction->stream(); in executeSendFunctionAsync() local
466 if (send_backward_stream) { in executeSendFunctionAsync()
470 if (send_backward_stream != default_stream) { in executeSendFunctionAsync()
473 send_backward_stream->wait(event); in executeSendFunctionAsync()