Searched refs:kAllGatherOpName (Results 1 – 11 of 11) sorted by relevance
46 if (op_name != kReduceScatter && op_name != kAllGatherOpName) { in GetKernelFormat()50 if (op_name == kAllGatherOpName && input_shape.size() <= kShape4dDims) { in GetKernelFormat()
33 {mindspore::kAllGatherOpName, mindspore::kHcomOpTypeAllGather},180 if (AnfAlgo::GetInputTensorNum(anf_node) > 1 && op_name == kAllGatherOpName && fusion >= 1) { in GetOutputSizeList()
164 if (AnfAlgo::GetCNodeName(anf_node) == kAllGatherOpName) { in GetHcomCount()
35 …if (AnfAlgo::GetCNodeName(cnode) == kAllGatherOpName && AnfAlgo::HasNodeAttr(kAttrFusion, cnode) && in Run()
129 auto prim = std::make_shared<Primitive>(kAllGatherOpName); in DefinePattern()
74 …Fusion(size_t groups = 1) : CommunicationOpFusion("all_gather_fusion", kAllGatherOpName, groups) {}
51 … (AnfAlgo::HasNodeAttr(kAttrRankSize, cnode) && AnfAlgo::GetCNodeName(cnode) == kAllGatherOpName) { in GenerateKernelBuildInfo()338 …:HasNodeAttr(kAttrRankSize, final_node) && AnfAlgo::GetCNodeName(final_node) == kAllGatherOpName) { in CreateFusedCommunicationOp()385 if (AnfAlgo::GetCNodeName(final_node) == kAllGatherOpName && is_recompute) { in CreateFusedCommunicationOp()
40 bool is_allgather = AnfAlgo::GetCNodeName(cnode) == kAllGatherOpName; in Run()
123 auto prim = std::make_shared<Primitive>(kAllGatherOpName); in DefinePattern()
68 constexpr auto kAllGatherOpName = "AllGather"; variable
1649 …td::set<std::string> kCommunicationOpNames = {kAllReduceOpName, kAllGatherOpName, kBroadcastOp… in IsCommunicationOp()