Searched defs:node_role (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/core/ |
| D | abstract_node.cc | 65 bool AbstractNode::Broadcast(const NodeRole &node_role, const DataPtr &message, size_t size, int co… in Broadcast()
177 bool AbstractNode::Send(const NodeRole &node_role, const uint32_t &rank_id, const DataPtr &data, si… in Send()
200 bool AbstractNode::Send(const NodeRole &node_role, const std::vector<uint32_t> &rank_ids, in Send()
240 bool AbstractNode::Send(const NodeRole &node_role, const uint32_t &rank_id, const DataPtr &message,… in Send()
280 bool AbstractNode::Send(const NodeRole &node_role, const std::vector<uint32_t> &rank_ids, in Send()
334 uint64_t AbstractNode::CollectiveSendAsync(const NodeRole &node_role, const uint32_t &rank_id, cons… in CollectiveSendAsync()
353 std::pair<uint32_t, uint64_t> AbstractNode::CollectiveReceiveAsync(const NodeRole &node_role, const… in CollectiveReceiveAsync()
|
| D | comm_util.cc | 138 bool CommUtil::ValidateRankId(const enum NodeRole &node_role, const uint32_t &rank_id, const int32_… in ValidateRankId()
|