Searched defs:switch_node (Results 1 – 11 of 11) sorted by relevance
180 auto switch_node = BlongToWhichSwitch(exit_node); in IdentifyWhileNodeOutput() local235 auto switch_node = BlongToWhichSwitch(exit_node); in UpdateExitNodeUser() local473 auto switch_node = input_node->cast<CNodePtr>(); in IdentifyBodySubgraphInput() local518 auto switch_node = BlongToWhichSwitch(next_iteration_cnode); in IdentifyBodySubgraphOutput() local
119 auto switch_node = input_node->cast<CNodePtr>(); in IdentifySubgraphInput() local
62 AnfNodePtr switch_node = nullptr; in CheckSwitchWithSideEffect() local
35 …auto switch_node = prim::GetPythonOps("geswitch", "mindspore.ops.functional")->cast<PrimitivePtr>(… in GenerateSwitchNode() local549 auto switch_node = inp0->cast<CNodePtr>(); in CheckSwitchWrapNode() local
839 auto switch_node = switch_call->input(primary_index); in operator ()() local
62 void FetchParameterBySwitchNode(const AnfNodePtr &switch_node, FuncGraphToParameter *graph_to_real_… in FetchParameterBySwitchNode()367 std::vector<AnfNodePtr> FetchOutputBySwitchNode(const AnfNodePtr &switch_node, std::set<AnfNodePtr>… in FetchOutputBySwitchNode()859 void ControlNodeParser::FetchValueNodeBySwitchNode(const AnfNodePtr &switch_node, in FetchValueNodeBySwitchNode()
404 int InferShapePass::InferSwitchNode(const std::unique_ptr<CNodeT> &switch_node, MetaGraphT *graph) { in InferSwitchNode()
503 …auto switch_node = kernel::LiteKernelUtil::GetInputsSpecificNode(node, schema::PrimitiveType_Switc… in GetSwitchAndCallNode() local
1647 int Scheduler::InferSwitchShape(const lite::LiteGraph::Node *switch_node) { in InferSwitchShape()
980 void SaveBranchCaller(const CNodePtr &switch_node, const FuncGraphPtr &branch) { in SaveBranchCaller()
810 auto switch_node = graph->NewCNode(switch_inputs); in CreateCallSwitchInputs() local