Searched refs:sendFunction (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/test/cpp/dist_autograd/ |
| D | test_dist_autograd.cpp | 94 auto sendFunction = context->retrieveSendFunction(0); in TEST_F() local
95 sendFunction->setGrads({t}); in TEST_F()
99 .executeSendFunctionAsync(context, sendFunction, /*retrainGraph*/ false) in TEST_F()
|
| /external/pytorch/torch/csrc/distributed/autograd/engine/ |
| D | dist_engine.cpp | 458 const std::shared_ptr<SendRpcBackward>& sendFunction, in executeSendFunctionAsync() argument
465 const auto& send_backward_stream = sendFunction->stream(); in executeSendFunctionAsync()
467 for (const auto& grad : sendFunction->getGrads()) { in executeSendFunctionAsync()
496 sendFunction, in executeSendFunctionAsync()
549 at::launch([this, graphTask, sendFunction]() { in executeSendFunctionAsync()
551 /*node_task*/ NodeTask(graphTask, sendFunction, InputBuffer(0)), in executeSendFunctionAsync()
|
| D | dist_engine.h | 49 const std::shared_ptr<SendRpcBackward>& sendFunction,
|
| /external/pytorch/torch/csrc/distributed/rpc/ |
| D | request_callback_no_python.cpp | 366 std::shared_ptr<SendRpcBackward> sendFunction = in processBackwardAutogradReq() local
370 sendFunction->setGrads(gradientsCall.getGrads()); in processBackwardAutogradReq()
374 autogradContext, sendFunction, gradientsCall.retainGraph()); in processBackwardAutogradReq()
|