Searched refs:recvFunctions (Results 1 – 3 of 3) sorted by relevance
46 auto recvFunctions = ctx.recvFunctions(); in dist_autograd_init() local50 for (const auto& map_entry : recvFunctions) { in dist_autograd_init()
50 std::unordered_map<int64_t, std::shared_ptr<RecvRpcBackward>> recvFunctions()
66 DistAutogradContext::recvFunctions() const { in recvFunctions() function in torch::distributed::autograd::DistAutogradContext