Searched refs:kPrimMerge (Results 1 – 3 of 3) sorted by relevance
101 auto fw_merge = VectorRef({GetPrim(prim::kPrimMerge), fw_merge_enter, fw_next_iter}); in DefineBidirectionRnnPattern()119 auto counter_merge1 = VectorRef({GetPrim(prim::kPrimMerge), counter_zero, counter_next_iter}); in DefineBidirectionRnnPattern()139 auto merge3 = VectorRef({GetPrim(prim::kPrimMerge), enter3, next_iteration3}); in DefineBidirectionRnnPattern()160 auto merge2 = VectorRef({GetPrim(prim::kPrimMerge), enter2, next_iter2}); in DefineBidirectionRnnPattern()
38 static bool IsMerge(const AnfNodePtr &node) { return CheckPrimitiveType(node, prim::kPrimMerge); } in IsMerge()
442 inline const PrimitivePtr kPrimMerge = std::make_shared<Primitive>("Merge"); variable