Home
last modified time | relevance | path

Searched defs:switch_cnode (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_cond.cc29 STATUS FunctionalizeCond::GetSwitchBranchType(const CNodePtr &switch_cnode, BranchType *branch_type… in GetSwitchBranchType()
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dmindir_control_flow_adjust.cc183 auto switch_cnode = GetMainFgSwitchNode(main_fg); in AddAfterFgForInlinedFg() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dpartial_eliminate.h351 …auto switch_cnode = old_cnode->func_graph()->NewCNode({input0_cnode->input(0), input0_cnode->input… in BuildNewSwitchNode() local
Dbranch_culling.cc562 auto switch_cnode = cnode->input(0)->cast<CNodePtr>(); in TransformSwitchBranchReplace() local
Dupdatestate_eliminate.cc844 auto switch_cnode = dyn_cast<CNode>(switch_node); in operator ()() local
Dincorporate_getitem.h698 auto switch_cnode = switch_->cast<CNodePtr>(); in ShrinkFuncGraphOutput() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dcontrol_flow_pass.cc485 auto switch_cnode = cond_fg->NewCNode(switch_node_inputs); in ProcessWhileOp() local
728 auto switch_cnode = fg->NewCNode(switch_node_inputs); in ProcessIfOp() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dcontrol_node_parser.cc63 const auto &switch_cnode = switch_node->cast<CNodePtr>(); in FetchParameterBySwitchNode() local
1223 const auto &switch_cnode = inputs[0]->cast<CNodePtr>(); in FetchFuncGraphToParameter() local
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dauto_monad.cc631 …EffectInfo TraceTupleFromSwitch(const CNodePtr &switch_cnode, const std::stack<int64_t> &tuple_ind… in TraceTupleFromSwitch()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc2260 auto switch_cnode = pre_cnode->input(0)->cast<CNodePtr>(); in FindLossCNode() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc784 auto switch_cnode = cnode_input->cast<CNodePtr>(); in CreateCallSwitchInputs() local