Searched refs:new_partial_inputs (Results 1 – 1 of 1) sorted by relevance
900 std::vector<AnfNodePtr> new_partial_inputs; in CreateCallSwitchLayerInputs() local908 new_partial_inputs = partial_node->inputs(); in CreateCallSwitchLayerInputs()910 …new_partial_inputs.emplace_back(NewValueNode(std::make_shared<Primitive>(prim::kPrimPartial->name(… in CreateCallSwitchLayerInputs()911 new_partial_inputs.emplace_back(partial_idx); in CreateCallSwitchLayerInputs()923 new_partial_inputs.insert(new_partial_inputs.end(), real_inputs.begin(), real_inputs.end()); in CreateCallSwitchLayerInputs()925 auto new_partial = graph->NewCNode(new_partial_inputs); in CreateCallSwitchLayerInputs()