Searched refs:CollectiveSendAsync (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | collective_ops_impl.cc | 72 …auto send_req_id = server_node_->CollectiveSendAsync(ps::core::NodeRole::SERVER, send_to_rank, sen… in RingAllReduce()
110 …auto send_req_id = server_node_->CollectiveSendAsync(ps::core::NodeRole::SERVER, send_to_rank, sen… in RingAllReduce()
175 …auto send_req_id = server_node_->CollectiveSendAsync(ps::core::NodeRole::SERVER, 0, sendbuff, coun… in ReduceBroadcastAllReduce()
189 … server_node_->CollectiveSendAsync(ps::core::NodeRole::SERVER, i, output_buff, count * sizeof(T)); in ReduceBroadcastAllReduce()
|
| /third_party/mindspore/mindspore/ccsrc/ps/core/ |
| D | abstract_node.h | 102 …uint64_t CollectiveSendAsync(const NodeRole &node_role, const uint32_t &rank_id, const void *data,…
|
| D | abstract_node.cc | 334 uint64_t AbstractNode::CollectiveSendAsync(const NodeRole &node_role, const uint32_t &rank_id, cons… in CollectiveSendAsync() function in mindspore::ps::core::AbstractNode
|