Searched refs:sendFunctions (Results 1 – 5 of 5) sorted by relevance
64 auto sendFunctions = ctx->sendFunctions(); in dist_autograd_init() local68 for (const auto& map_entry : sendFunctions) { in dist_autograd_init()
214 auto sendFunctions = autogradContext->sendFunctions(); in computeDependencies() local217 for (const auto& mapEntry : sendFunctions) { in computeDependencies()299 for (const auto& mapEntry : sendFunctions) { in computeDependencies()
45 auto send_function = autogradContext->sendFunctions()[1]; in TEST_F()
46 std::unordered_map<int64_t, std::shared_ptr<SendRpcBackward>> sendFunctions()
60 DistAutogradContext::sendFunctions() const { in sendFunctions() function in torch::distributed::autograd::DistAutogradContext