Home
last modified time | relevance | path

Searched refs:set_default_param (Results 1 – 25 of 26) sorted by relevance

12

/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dinfershape_pass.cc276 param_node->set_default_param(cnode->input(index)->cast<ParameterPtr>()->default_param()); in SetSubGraphInput()
288 param_node->set_default_param(tensor_info); in SetSubGraphInput()
293 param_node->set_default_param(tensor_info); in SetSubGraphInput()
397 sub_param_input->set_default_param(nullptr); in ResetSubGraphInput()
Ddecrease_transpose_algo.cc509 param_node->set_default_param(cnode->input(index)->cast<ParameterPtr>()->default_param()); in SetSubGraphInput()
519 …param_node->set_default_param(std::make_shared<tensor::Tensor>((TypeId)data_info.data_type_, shape… in SetSubGraphInput()
521 …param_node->set_default_param(std::make_shared<tensor::Tensor>((TypeId)data_info.data_type_, shape… in SetSubGraphInput()
543 sub_param_input->set_default_param(nullptr); in ResetSubGraphInput()
/third_party/mindspore/mindspore/lite/test/ut/tools/converter/registry/parser/
Dmodel_parser_test.cc124 parameter->set_default_param(tensor_info); in BuildGraphNodes()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc56 cache_param->set_default_param(MakeValue(new_tensor)); in AddCacheParameters()
260 hashmap->set_default_param(MakeValue(new_tensor)); in InitHashMap()
273 step->set_default_param(MakeValue(new_tensor)); in InitStep()
540 new_param->set_default_param(MakeValue(new_tensor)); in CreateOutputNodeParam()
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_model_parser.cc93 const_node->set_default_param(tensor_info); in CreateConstParamter()
205 parameter_node->set_default_param(tensor_info); in BuildParameterNode()
885 ext_subgraph_input->set_default_param(copy_tensor_info); in BuildCNode()
1211 while_tensor_array_input->set_default_param(tensor_info); in AddTensorArrayEdge()
1301 parameter_node->set_default_param(tensor_info); in BuildParameterNodeForQuantParam()
/third_party/mindspore/tests/ut/cpp/session/
Dkernel_graph_test.cc86 weight_parameter_node->set_default_param(param_value_new); in TEST_F()
Danf_runtime_algorithm_test.cc768 parameter_node->set_default_param(param_value_new); in TEST_F()
/third_party/mindspore/mindspore/lite/tools/common/
Dtensor_util.cc192 param_node->set_default_param(tensor_info); in InitParameterFromTensorInfo()
/third_party/mindspore/mindspore/lite/tools/converter/
Dexport_model.cc118 mirror_parameter->set_default_param(tensor_info); in CloneParameterAndValueNode()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dgraph_partitioner.cc280 dst->set_default_param(src->default_param()); in CreateNewGraph()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dpy_pass.cc226 param_node->set_default_param(param_value); in ReflectParamBackToPython()
/third_party/mindspore/mindspore/core/ir/
Dfunc_graph_cloner.cc79 new_param->set_default_param(old_param->default_param()); in CloneParameter()
283 param->set_default_param(old_param->default_param()); in CloneParameter()
Dfunc_graph.cc727 parameter->set_default_param(MakeValue(meta_tensor)); in add_weight()
Danf.h404 void set_default_param(ValuePtr param) { in set_default_param() function
/third_party/mindspore/mindspore/lite/tools/converter/acl/
Dacl_pass.cc71 om_parameter->set_default_param(param_value); in CreateOmParameter()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dresolve.cc126 node->set_default_param(value); in ResolveParameterObj()
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconv_transform_fusion.cc260 new_weight_paramter->set_default_param(new_weight_tensor); in GenNewConvTensor()
Dtf_bidirection_gru_fusion.cc451 parameter->set_default_param(gate_weight_default); in AddDefaultParameter()
/third_party/mindspore/mindspore/core/load_mindir/
Danf_model_parser.cc272 node->set_default_param(load_tensor_info); in BuildParameterForFuncGraph()
294 node->set_default_param(tensor_info); in BuildParameterForFuncGraph()
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Ddepthwise_conv2d_info.cc437 parameter_node->set_default_param(split_constant_tensor); in CreateConstantOutputsOfSplit()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dkpynative.cc999 p->set_default_param(input_w->default_param()); in SetSensAndWeights()
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc951 param_node->set_default_param(tensor_info_new); in BuildParameterNode()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline.cc1118 param_cast->set_default_param(iter->second); in UpdataParamNodeDefaultInput()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dkernel_graph.cc570 new_parameter->set_default_param(parameter->default_param()); in NewParameter()
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute.cc1373 free_param->set_default_param(value); in GetInput()
2614 free_param->set_default_param(value); in GetWeightsArgs()

12