Searched refs:CollectiveReceiveAsync (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | collective_ops_impl.cc | 82 …auto recv_req_id = server_node_->CollectiveReceiveAsync(ps::core::NodeRole::SERVER, recv_from_rank… in RingAllReduce()
119 …auto recv_req_id = server_node_->CollectiveReceiveAsync(ps::core::NodeRole::SERVER, recv_from_rank… in RingAllReduce()
159 … auto recv_req_id = server_node_->CollectiveReceiveAsync(ps::core::NodeRole::SERVER, i, &recv_str); in ReduceBroadcastAllReduce()
198 … auto recv_req_id = server_node_->CollectiveReceiveAsync(ps::core::NodeRole::SERVER, 0, &recv_str); in ReduceBroadcastAllReduce()
|
| /third_party/mindspore/mindspore/ccsrc/ps/core/ |
| D | abstract_node.h | 103 …std::pair<uint32_t, uint64_t> CollectiveReceiveAsync(const NodeRole &node_role, const uint32_t &ra…
|
| D | abstract_node.cc | 353 std::pair<uint32_t, uint64_t> AbstractNode::CollectiveReceiveAsync(const NodeRole &node_role, const… in CollectiveReceiveAsync() function in mindspore::ps::core::AbstractNode
|