Searched refs:kPrimMiniStepAllGather (Results 1 – 5 of 5) sorted by relevance
45 … std::set<PrimitivePtr> ALLGATHER_NODE_LIST_ = {prim::kPrimAllGather, prim::kPrimMiniStepAllGather,
107 … "mini_step_allgather_replace", prim::kPrimMiniStepAllGather); in OptimizeIRPassLib()
278 if (!IsPrimitiveCNode(node, prim::kPrimMiniStepAllGather) || node->func_graph() == nullptr) { in operator()
412 inline const PrimitivePtr kPrimMiniStepAllGather = std::make_shared<Primitive>("_MiniStepAllGather"… variable
2611 …itiveCNode(cnode, prim::kPrimAllGather) || IsPrimitiveCNode(cnode, prim::kPrimMiniStepAllGather) || in IsCohesiveNode()