Home
last modified time | relevance | path

Searched refs:new_parameter (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dcontrol_flow_pass.cc265 auto new_parameter = cond_fg->add_parameter(); in CreateWhileCondCallNode() local
266 MS_CHECK_TRUE_MSG(new_parameter != nullptr, lite::RET_NULL_PTR, "new_parameter is nullptr"); in CreateWhileCondCallNode()
267 new_parameter->set_name(item->fullname_with_scope() + "_cond_fg_parameter"); in CreateWhileCondCallNode()
268 new_parameter->set_abstract(item->abstract()); in CreateWhileCondCallNode()
269 (*visited_nodes_and_cond_fg_inputs_replace_pairs)[item] = new_parameter; in CreateWhileCondCallNode()
270 cond_nodes_used_by_after_partial->push_back(new_parameter); in CreateWhileCondCallNode()
318 auto new_parameter = body_fg->add_parameter(); in CreateWhileBodyPartialNode() local
319 MS_CHECK_TRUE_MSG(new_parameter != nullptr, lite::RET_NULL_PTR, "new_parameter is nullptr"); in CreateWhileBodyPartialNode()
320 new_parameter->set_name(cond_fg_inputs[i]->fullname_with_scope() + "_body_fg_parameter"); in CreateWhileBodyPartialNode()
321 new_parameter->set_abstract(cond_fg_inputs[i]->abstract()); in CreateWhileBodyPartialNode()
[all …]
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_pad_adjust.cc68 auto new_parameter = CreateNewParameter(func_graph, shape); in NewReshapeOpNode() local
69 MS_CHECK_TRUE_MSG(new_parameter != nullptr, nullptr, "create parameter return nullptr"); in NewReshapeOpNode()
70 new_parameter->set_name(input_node->fullname_with_scope() + "_reshape/shape"); in NewReshapeOpNode()
71 std::vector<AnfNodePtr> op_inputs = {value_node, input_node, new_parameter}; in NewReshapeOpNode()
90 auto new_parameter = CreateNewParameter(func_graph, perm); in NewTransposeOpNode() local
91 MS_CHECK_TRUE_MSG(new_parameter != nullptr, nullptr, "create parameter return nullptr"); in NewTransposeOpNode()
92 new_parameter->set_name(input_node->fullname_with_scope() + "_transpose/perm"); in NewTransposeOpNode()
93 std::vector<AnfNodePtr> op_inputs = {value_node, input_node, new_parameter}; in NewTransposeOpNode()
/third_party/mindspore/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc505 auto new_parameter = sub_graph->add_parameter(); in CreateParameterForPartialSubGraph() local
506 new_parameter->set_name(graph_name + "_input_" + input->fullname_with_scope()); in CreateParameterForPartialSubGraph()
507 new_parameter->set_abstract(input->abstract()); in CreateParameterForPartialSubGraph()
508 (*partial_inputs_and_subgraph_input_map)[input] = new_parameter; in CreateParameterForPartialSubGraph()
518 auto new_parameter = sub_graph->add_parameter(); in CreateParameterForPartialSubGraph() local
519 new_parameter->set_name(graph_name + "_input_" + input->fullname_with_scope()); in CreateParameterForPartialSubGraph()
520 new_parameter->set_abstract(input->abstract()); in CreateParameterForPartialSubGraph()
521 (*partial_inputs_and_subgraph_input_map)[input] = new_parameter; in CreateParameterForPartialSubGraph()
522 partial_inputs->emplace_back(new_parameter); in CreateParameterForPartialSubGraph()
528 auto new_parameter = sub_graph->add_parameter(); in CreateParameterForPartialSubGraph() local
[all …]
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconstant_folding_fusion.cc236 auto new_parameter = CreateNewParamter(func_graph, output_tensors.at(k)); in ReplaceCNode() local
237 if (new_parameter == nullptr) { in ReplaceCNode()
241 new_parameter->set_name(input_node->fullname_with_scope() + "_const_" + std::to_string(k)); in ReplaceCNode()
242 (void)manager->Replace(tuple_node, new_parameter); in ReplaceCNode()
249 auto new_parameter = CreateNewParamter(func_graph, output_tensors.front()); in ReplaceCNode() local
250 if (new_parameter == nullptr) { in ReplaceCNode()
254 new_parameter->set_name("constfold_" + input_node->fullname_with_scope()); in ReplaceCNode()
255 (void)manager->Replace(input_node, new_parameter); in ReplaceCNode()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_json_decoder.cc339 ParameterPtr new_parameter = func_graph->add_parameter(); in DecodeParameter() local
341 new_parameter->set_name(name); in DecodeParameter()
346 new_parameter->set_abstract(abstract); in DecodeParameter()
348 new_parameter->set_kernel_info(kernel_info); in DecodeParameter()
352 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), new_parameter.get()); in DecodeParameter()
353 nodes_map_[name] = new_parameter; in DecodeParameter()
354 return new_parameter; in DecodeParameter()
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dmindir_control_flow_adjust.cc122 auto new_parameter = after_fg->add_parameter(); in AddAfterFuncGraph() local
123 MS_CHECK_TRUE_MSG(new_parameter != nullptr, nullptr, "new_parameter is nullptr."); in AddAfterFuncGraph()
124 new_parameter->set_name(switch_node_name + ":" + std::to_string(i++)); in AddAfterFuncGraph()
125 new_parameter->set_abstract(iter->abstract()); in AddAfterFuncGraph()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dkernel_graph.cc565 auto new_parameter = NewParameter(abstract); in NewParameter() local
568 new_parameter->set_name(parameter->name()); in NewParameter()
570 new_parameter->set_default_param(parameter->default_param()); in NewParameter()
574 SetKernelInfoForNode(new_parameter); in NewParameter()
575 AnfAlgo::SetGraphId(graph_id_, new_parameter.get()); in NewParameter()
576 return new_parameter; in NewParameter()
580 ParameterPtr new_parameter = add_parameter(); in NewParameter() local
581 new_parameter->set_abstract(abstract); in NewParameter()
583 SetKernelInfoForNode(new_parameter); in NewParameter()
584 AnfAlgo::SetGraphId(graph_id_, new_parameter.get()); in NewParameter()
[all …]
Dsession_basic.cc593 auto new_parameter = graph->TransTupleToMakeTuple(graph->NewParameter(node->abstract())); in CreateParameterFromTuple() local
594 auto parameters = AnfAlgo::GetAllOutput(new_parameter); in CreateParameterFromTuple()
624 return new_parameter; in CreateParameterFromTuple()
638 ParameterPtr new_parameter = nullptr; in CreateNewParameterFromParameter() local
641 new_parameter = param_value->parameter(); in CreateNewParameterFromParameter()
643 if (new_parameter == nullptr) { in CreateNewParameterFromParameter()
645 new_parameter = graph->NewParameter(anf->cast<ParameterPtr>()); in CreateNewParameterFromParameter()
649 InitInternalOutputParameter(input_node_iter->second, new_parameter); in CreateNewParameterFromParameter()
653 param_value->set_parameter(new_parameter); in CreateNewParameterFromParameter()
656 new_parameter->IncreaseUsedGraphCount(); in CreateNewParameterFromParameter()
[all …]
Dcpu_session.cc71 ParameterPtr new_parameter = graph->NewParameter(anf->cast<ParameterPtr>()); in CreateNewParameterFromParameter() local
73 graph_inputs->push_back(new_parameter); in CreateNewParameterFromParameter()
75 return new_parameter; in CreateNewParameterFromParameter()
Dkernel_graph.h123 void ReplaceGraphInput(const AnfNodePtr &old_parameter, const AnfNodePtr &new_parameter);
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dadd_stitch_atomic_clean_gpu.cc34 … const AnfNodePtr &new_parameter) const { in CreateInplaceAssignNode()
38 …CreateCNode({NewValueNode(prim::kPrimInplaceAssign), new_parameter, atomic_add_node_, out_node}, s… in CreateInplaceAssignNode()
Dadd_stitch_atomic_clean_gpu.h37 …CNodePtr CreateInplaceAssignNode(const FuncGraphPtr &sub_graph, const AnfNodePtr &new_parameter) c…
Dadd_atomic_clean.h87 …eInplaceAssignNodeAndCorrectReturn(const FuncGraphPtr &sub_graph, const AnfNodePtr &new_parameter);
Dadd_atomic_clean.cc272 … const AnfNodePtr &new_parameter) { in CreateInplaceAssignNodeAndCorrectReturn() argument
293 …CreateCNode({NewValueNode(prim::kPrimInplaceAssign), new_parameter, atomic_add_node_, out_node}, s… in CreateInplaceAssignNodeAndCorrectReturn()
Dgraph_kernel_helper.cc944 AnfNodePtrList new_parameter, new_inputs; in EliminateRedundantParameters() local
947 new_parameter.push_back(ori_parameter[i]); in EliminateRedundantParameters()
951 func_graph->set_parameters(new_parameter); in EliminateRedundantParameters()
/third_party/mindspore/mindspore/boost/
Dboost_cell_wrapper.py41 def _get_delta_weight_process(new_parameter, old_parameter): argument
42 delta_w = old_parameter - new_parameter
50 def _save_weight_process(new_parameter, old_parameter): argument
51 return P.Assign()(new_parameter, old_parameter)
Dadasum.py41 new_parameter = old_parameter - update_delta_weight
42 return P.Assign()(parameter, new_parameter)
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dpy_pass.cc214 …py::object new_parameter = parameter_class(default_tensor, param_name, requires_grad, layerwise_pa… in ReflectParamBackToPython() local
216 top_cell.attr(SET_PARAM)(param_name, new_parameter); in ReflectParamBackToPython()
219 bool converted = parse::ConvertData(new_parameter, &param_value, false); in ReflectParamBackToPython()