Searched refs:switch_layer_inputs (Results 1 – 3 of 3) sorted by relevance
163 auto switch_layer_inputs = swich_layer_cnode->inputs(); in operator() local164 if (switch_layer_inputs.size() != 3) { in operator()169 …bool layer_changed = TransformLayerNode(switch_layer_inputs[2], optimizer->manager(), &transformed… in operator()171 switch_layer_inputs[2] = transformed; in operator()172 call_inputs[0] = switch_layer_call_cnode->func_graph()->NewCNode(switch_layer_inputs); in operator()
90 const auto &switch_layer_inputs = switch_layer_cnode->inputs(); in FetchParameterBySwitchLayerNode() local92 if (switch_layer_inputs.size() != kSwitchLayerInputNum) { in FetchParameterBySwitchLayerNode()96 auto tuple_inputs = switch_layer_inputs[kSwitchLayerBranchPos]->cast<CNodePtr>()->inputs(); in FetchParameterBySwitchLayerNode()533 const auto &switch_layer_inputs = inputs[0]->cast<CNodePtr>()->inputs(); in IsSubCallNode() local534 const auto tuple_inputs = switch_layer_inputs[kSwitchLayerBranchPos]->cast<CNodePtr>()->inputs(); in IsSubCallNode()
883 std::vector<AnfNodePtr> switch_layer_inputs = {switch_layer_cnode->input(kAnfPrimitiveIndex), in CreateCallSwitchLayerInputs() local934 switch_layer_inputs.emplace_back(new_make_tuple); in CreateCallSwitchLayerInputs()935 auto new_switch_layer = graph->NewCNode(switch_layer_inputs); in CreateCallSwitchLayerInputs()