Searched refs:kPrimLabelSwitch (Results 1 – 6 of 6) sorted by relevance
121 if (IsPrimitiveCNode(cnode, prim::kPrimLabelSwitch)) { in AssignLabelForGotoSwitch()
527 if (IsPrimitiveCNode(kernel_node, prim::kPrimLabelSwitch)) { in SelectKernelInfo()
799 AnfAlgo::CheckPrimitiveType(*iter, prim::kPrimLabelSwitch) || in SetIterEndAttr()1365 auto label_switch = NewPrimitive(prim::kPrimLabelSwitch); in LabelSwitch()1527 if (AnfAlgo::CheckPrimitiveType(node, prim::kPrimLabelSwitch)) { in UnfoldRepeatedLabels()1826 if (IsPrimitiveCNode(node, prim::kPrimLabelSwitch)) { in EraseLabel()
1431 if (AnfAlgo::CheckPrimitiveType(kernel_cnodes[i], prim::kPrimLabelSwitch) && in InitMemReuseExecOrder()
1925 return IsPrimitive(input, prim::kPrimLabelGoto) || IsPrimitive(input, prim::kPrimLabelSwitch); in IsCondControlKernel()
156 inline const PrimitivePtr kPrimLabelSwitch = std::make_shared<Primitive>("LabelSwitch"); variable