| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tf/ |
| D | functionalize_while.cc | 185 auto switch_node = BlongToWhichSwitch(exit_node); in IdentifyWhileNodeOutput() local
242 auto switch_node = BlongToWhichSwitch(exit_node); in UpdateExitNodeUser() local
482 auto switch_node = input_node->cast<CNodePtr>(); in IdentifyBodySubgraphInput() local
528 auto switch_node = BlongToWhichSwitch(next_iteration_cnode); in IdentifyBodySubgraphOutput() local
|
| D | remove_ineffective_control_flow.cc | 65 auto switch_node = cnode->input(1); in CheckIfIneffective() local
|
| D | functionalize_cond.cc | 151 auto switch_node = input_node->cast<CNodePtr>(); in IdentifySubgraphInput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | partial_eliminate.h | 325 auto switch_node = switch_call->input(0)->cast<CNodePtr>(); in operator() local
358 const auto switch_node = input0->cast<CNodePtr>(); in BuildNewSwitchNode() local
|
| D | branch_culling.cc | 45 …auto switch_node = prim::GetPythonOps("geswitch", "mindspore.ops.functional")->cast<PrimitivePtr>(… in GenerateSwitchNode() local
600 auto switch_node = inp0->cast<CNodePtr>(); in CheckSwitchWrapNode() local
|
| D | updatestate_eliminate.cc | 952 auto switch_node = switch_call->input(primary_index); in operator ()() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/ |
| D | switch_not_cut.cc | 243 const auto &switch_node = cnode->input(0)->cast<CNodePtr>(); in Run() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/runtime/graph_scheduler/ |
| D | control_node_parser_test.cc | 93 auto switch_node = root_func_graph->NewCNode(switch_inputs); in BuildGraphs() local
|
| D | graph_scheduler_test.cc | 75 auto switch_node = root_func_graph->NewCNode(switch_inputs); in BuildGraphs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/actor/ |
| D | switch_actor.cc | 71 …auto switch_node = kernel::KernelExecUtil::GetInputsSpecificNode(node, schema::PrimitiveType_Switc… in GetSwitchAndCallNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/ |
| D | utils.cc | 230 CNodePtr switch_node = nullptr; in IsIfNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_compiler/ |
| D | func_graph_builder.cc | 291 CNodePtr switch_node = in Mutate_() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/static_analysis/ |
| D | auto_monad.cc | 631 const auto &switch_node = caller->input(0); in CheckAndFixSwitchCall() local
1531 void SaveBranchCaller(const CNodePtr &switch_node, const FuncGraphVector &branches) { in SaveBranchCaller()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 632 auto switch_node = GetNodeInput<CNode>(lazy_call_node, kZeroNum); in FindLoopSwitchNode() local
639 FuncGraphPtr GetSubgraph(const CNodePtr &switch_node, size_t input_index, const std::string &name_p… in GetSubgraph()
707 auto switch_node = FindLoopSwitchNode(control_subgraph); in TraceLoopToControlMap() local
724 auto switch_node = FindLoopSwitchNode(control_subgraph); in TraceAfterToLoopMap() local
783 auto switch_node = FindLoopSwitchNode(control_subgraph); in MatchGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/ |
| D | scheduler.cc | 1754 int Scheduler::InferSwitchShape(const lite::LiteGraph::Node *switch_node) { in InferSwitchShape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/ |
| D | kernel_graph_mgr.cc | 1431 auto switch_node = graph->NewCNode(switch_inputs); in CreateCallSwitchInputs() local
|