Home
last modified time | relevance | path

Searched defs:switch_input (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_cond.cc153 auto switch_input = utils::cast<CNodePtr>(switch_node->input(1)); in IdentifySubgraphInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/c_api/src/
Dnode.cc164 …ePtr BuildSwitchStructure(ResMgrHandle res_mgr, GraphHandle graph, NodeHandle const switch_input[], in BuildSwitchStructure()
248 NodeHandle switch_input[] = {cond_raw_ptr, true_raw_ptr, false_raw_ptr}; in MSNewSwitch() local
356 NodeHandle switch_input[] = {cond_raw_ptr, body_raw_ptr, after_raw_ptr}; in MSNewWhile() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dscheduler.cc856 auto switch_input = NodeInputIsSwitchType(node); in InferCallShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dkernel_graph_mgr.cc1686 auto switch_input = CreateSwitchInput(cnode, node_input, graph); in CreateCNodeInputs() local