Searched refs:kAllReduceOpName (Results 1 – 9 of 9) sorted by relevance
36 if (kernel_name == kAllReduceOpName) { in AssignGpuStream()72 if (kernel_name == kAllReduceOpName) { in FindAllReduceStreamSwitchPos()78 } else if (AnfAlgo::GetCNodeName(*mock_send_node_iter) != kAllReduceOpName) { in FindAllReduceStreamSwitchPos()93 } else if (AnfAlgo::GetCNodeName(*mock_recv_node_iter) != kAllReduceOpName) { in FindAllReduceStreamSwitchPos()134 if (AnfAlgo::GetCNodeName(node) != kAllReduceOpName) { in FindRecvNodePos()223 if (AnfAlgo::GetCNodeName(mock_send_node) == kAllReduceOpName) { in CacheSendRecvCNodesForAllReduce()227 if (AnfAlgo::GetCNodeName(mock_recv_node) == kAllReduceOpName) { in CacheSendRecvCNodesForAllReduce()
65 …if ((node->isa<CNode>() && AnfAlgo::GetCNodeName(node) == kAllReduceOpName) || node->isa<Parameter… in TEST_F()106 …if ((node->isa<CNode>() && AnfAlgo::GetCNodeName(node) == kAllReduceOpName) || node->isa<Parameter… in TEST_F()148 …if ((node->isa<CNode>() && AnfAlgo::GetCNodeName(node) == kAllReduceOpName) || node->isa<Parameter… in TEST_F()152 if (node->isa<CNode>() && AnfAlgo::GetCNodeName(node) == kAllReduceOpName) { in TEST_F()204 …if ((node->isa<CNode>() && AnfAlgo::GetCNodeName(node) == kAllReduceOpName) || node->isa<Parameter… in TEST_F()
68 …Fusion(size_t groups = 1) : CommunicationOpFusion("all_reduce_fusion", kAllReduceOpName, groups) {}
35 const std::set<std::string> kInOutOperatorSet = {kAllReduceOpName, kBroadcastOpName, kMakeTupleName…
216 …dePtr> allreduce_inputs = {NewValueNode(std::make_shared<Primitive>(kAllReduceOpName)), input_node… in CreateAllReduceAndMul()
32 {mindspore::kAllReduceOpName, mindspore::kHcomOpTypeAllReduce},
67 constexpr auto kAllReduceOpName = "AllReduce"; variable
838 if (node_name == kAllReduceOpName || node_name == kBroadcastOpName) { in GetAttrGroups()
1649 …static const std::set<std::string> kCommunicationOpNames = {kAllReduceOpName, kAllGatherOpName… in IsCommunicationOp()