Home
last modified time | relevance | path

Searched defs:switch_layer_inputs (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dcall_graph_tuple_transform.h163 auto switch_layer_inputs = swich_layer_cnode->inputs(); in operator() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dcontrol_node_parser.cc90 const auto &switch_layer_inputs = switch_layer_cnode->inputs(); in FetchParameterBySwitchLayerNode() local
533 const auto &switch_layer_inputs = inputs[0]->cast<CNodePtr>()->inputs(); in IsSubCallNode() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc883 std::vector<AnfNodePtr> switch_layer_inputs = {switch_layer_cnode->input(kAnfPrimitiveIndex), in CreateCallSwitchLayerInputs() local