Home
last modified time | relevance | path

Searched refs:ParameterPtr (Results 1 – 25 of 194) sorted by relevance

12345678

/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconv_conv_fusion.cc56 STATUS GenNewConvBias(const ParameterPtr &down_bias_node, const ParameterPtr &down_weight_node, in GenNewConvBias()
57 const ParameterPtr &up_bias_node, const ParameterPtr &new_bias_node) { in GenNewConvBias()
117 STATUS GenNewConvWeight(const ParameterPtr &down_weight_node, const ParameterPtr &up_weight_node, in GenNewConvWeight()
118 const ParameterPtr &new_weight_node) { in GenNewConvWeight()
173 auto down_weight_parameter = down_conv_cnode->input(kConvWeightIndex)->cast<ParameterPtr>(); in ReplaceParametersAndNodes()
174 auto up_weight_parameter = up_conv_cnode->input(kConvWeightIndex)->cast<ParameterPtr>(); in ReplaceParametersAndNodes()
183 ParameterPtr new_bias_parameter{nullptr}; in ReplaceParametersAndNodes()
185 ParameterPtr down_bias_parameter; in ReplaceParametersAndNodes()
187 down_bias_parameter = down_conv_cnode->input(kConvBiasIndex)->cast<ParameterPtr>(); in ReplaceParametersAndNodes()
189 auto up_bias_parameter = up_conv_cnode->input(kConvBiasIndex)->cast<ParameterPtr>(); in ReplaceParametersAndNodes()
[all …]
Dtf_lstm_cell_fusion.h39 …e::STATUS SplitWeights(const AnfNodePtr &weight, const ParameterPtr &weight_i, const ParameterPtr …
41 …static lite::STATUS SetWeightAbstractAndDefault(const ParameterPtr &weight, const std::vector<int6…
43 …lite::STATUS PopulateBiasNode(const EquivPtr &body_equiv, const ParameterPtr &new_bias, const AnfN…
Dconv_bn_fusion.cc49 auto bn_var_param = bn_var_node->cast<ParameterPtr>()->default_param(); in CalTransale()
72 auto bn_scale_param = bn_scale_node->cast<ParameterPtr>()->default_param(); in CalTransale()
88 auto bn_mean_param = bn_mean_node->cast<ParameterPtr>()->default_param(); in CalTransBias()
100 auto bn_bias_param = bn_bias_node->cast<ParameterPtr>()->default_param(); in CalTransBias()
122 auto origin_param = origin_node->cast<ParameterPtr>()->default_param(); in CalEstimatedData()
128 auto scale_factor_param = scale_factor_node->cast<ParameterPtr>()->default_param(); in CalEstimatedData()
Dtf_lstm_cell_fusion.cc41 bool IsParameterNode(const BaseRef &n) { return utils::isa<ParameterPtr>(n); } in IsParameterNode()
185 STATUS TfLstmCellFusion::SetWeightAbstractAndDefault(const ParameterPtr &weight, const std::vector<… in SetWeightAbstractAndDefault()
222 STATUS TfLstmCellFusion::SplitWeights(const AnfNodePtr &weight, const ParameterPtr &weight_i, in SplitWeights()
223 const ParameterPtr &weight_c, int hidden_size) { in SplitWeights()
229 if (!utils::isa<ParameterPtr>(weight)) { in SplitWeights()
232 auto weight_param = utils::cast<ParameterPtr>(weight); in SplitWeights()
276 STATUS TfLstmCellFusion::PopulateBiasNode(const EquivPtr &body_equiv, const ParameterPtr &new_bias, in PopulateBiasNode()
281 if (!utils::isa<ParameterPtr>(old_bias)) { in PopulateBiasNode()
285 auto old_bias_param = utils::cast<ParameterPtr>(old_bias); in PopulateBiasNode()
378 if (!utils::isa<ParameterPtr>(hidden)) { in CreateLSTMNode()
[all …]
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dgllo_utils.h67 ParameterPtr AddNewBiasNode(float *bias_data, const FuncGraphPtr &func_graph, int kernel_num, TypeI…
93 ParameterPtr BuildParameterNode(const FuncGraphPtr &func_graph, const AnfNodePtr &node,
96 ParameterPtr BuildIntValueParameterNode(const FuncGraphPtr &func_graph, const int32_t &data,
99 ParameterPtr BuildIntVecParameterNode(const FuncGraphPtr &func_graph, const std::vector<int32_t> &d…
102 ParameterPtr BuildIntVec2DParameterNode(const FuncGraphPtr &func_graph, const std::vector<std::vect…
105 ParameterPtr BuildFloatValueParameterNode(const FuncGraphPtr &func_graph, const float &data,
108 ParameterPtr BuildFloatVecParameterNode(const FuncGraphPtr &func_graph, const std::vector<float> &d…
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.h86 std::map<std::string, mindspore::ParameterPtr> *switch_loop_input);
88 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
96 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
110 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input);
114 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input);
124 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
135 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
138 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
145 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
157 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input);
Dkernel_adjust.cc138 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input, in InsertIndepentParallel()
154 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input, in InsertFpBpLoopStreamSwitch()
235 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input, in InsertNextLoopAssignAdd()
248 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input) { in InsertCurrentLoopAssignAdd()
271 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input) { in InsertSwitchLoopInput()
283 ParameterPtr param_ptr = input->cast<ParameterPtr>(); in InsertSwitchLoopInput()
294 const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input) { in InsertGetNextLoopStreamSwitch()
357 … const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input, in InsertEosStreamSwitch()
434 std::map<std::string, mindspore::ParameterPtr> switch_loop_input; in InsertSwitchLoop()
515 … std::map<std::string, mindspore::ParameterPtr> *switch_loop_input) { in CreateSwitchOpParameters()
[all …]
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dfunction_block.h56 void SetPhiArgument(const ParameterPtr &phi);
57 bool CollectRemovablePhi(const ParameterPtr &phi);
63 AnfNodePtr SearchReplaceNode(const std::string &var, const ParameterPtr &phi);
78 …const std::unordered_map<ParameterPtr, AnfNodePtr> &removable_phis() const { return removable_phis… in removable_phis()
123 std::map<ParameterPtr, std::string> phi_nodes_;
130 std::unordered_map<ParameterPtr, AnfNodePtr> removable_phis_;
136 std::unordered_map<AnfNodePtr, ParameterPtr> resolve_to_removable_phis_;
/third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/
Dspecialize_test.cc60 ParameterPtr y = graph_g_->add_parameter(); in SetUp()
73 ParameterPtr x = graph_f_->add_parameter(); in SetUp()
97 ParameterPtr x1 = graph_beta_->add_parameter(); in SetUp()
164 ParameterPtr x = graph_g->add_parameter(); in GenerateFromTypes()
165 ParameterPtr y = graph_g->add_parameter(); in GenerateFromTypes()
193 ParameterPtr x = graph_->add_parameter(); in SetUp()
194 ParameterPtr y = graph_->add_parameter(); in SetUp()
Dstatic_analysis_test.cc98 ParameterPtr x = fg->add_parameter(); in GenerateFromTypes()
99 ParameterPtr y = fg->add_parameter(); in GenerateFromTypes()
136 ParameterPtr x = func_graph->add_parameter(); in MakeFuncGraph()
137 ParameterPtr y = func_graph->add_parameter(); in MakeFuncGraph()
193 ParameterPtr y = graph_g_->add_parameter(); in SetUp()
206 ParameterPtr x = graph_f_->add_parameter(); in SetUp()
230 ParameterPtr x1 = graph_beta_->add_parameter(); in SetUp()
333 ParameterPtr x = func_graph_->add_parameter(); in SetUp()
334 ParameterPtr y = func_graph_->add_parameter(); in SetUp()
/third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/
Dgather_fusion_mapper.cc38 if (!utils::isa<ParameterPtr>(axis_input)) { in Mapper()
42 ParameterPtr axis_param = axis_input->cast<ParameterPtr>(); in Mapper()
Dcast_mapper.cc38 if (!utils::isa<ParameterPtr>(to_input)) { in Mapper()
42 ParameterPtr to_param = to_input->cast<ParameterPtr>(); in Mapper()
Dtranspose_mapper.cc40 if (!utils::isa<ParameterPtr>(perm_input)) { in Mapper()
44 ParameterPtr perm_param = perm_input->cast<ParameterPtr>(); in Mapper()
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_inputs_adjust.cc72 STATUS ReplaceInt64ParameterNode(const FuncGraphPtr &func_graph, const ParameterPtr &param_node) { in ReplaceInt64ParameterNode()
116 bool ValidParameterNode(const ParameterPtr &param_node) { in ValidParameterNode()
178 auto param_node = anf_node->cast<ParameterPtr>(); in ReplaceTransposeWithGraphInput()
199 auto perm_param = perm_anf->cast<ParameterPtr>(); in ReplaceTransposeWithGraphInput()
252 const auto &param_node = cnode->input(opt::kInputIndexTwo)->cast<ParameterPtr>(); in AdjustStridedSlice()
335 … scale_node->isa<Parameter>() ? scale_node->cast<ParameterPtr>() : size_node->cast<ParameterPtr>(); in AdjustResize()
343 auto scale_param = scale_node->cast<ParameterPtr>(); in AdjustResize()
344 auto size_param = size_node->cast<ParameterPtr>(); in AdjustResize()
372 if (utils::isa<ParameterPtr>(node)) { in Adjust()
373 auto param_node = node->cast<ParameterPtr>(); in Adjust()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_inference_session.cc42 auto pk_node = input_nodes[i]->cast<ParameterPtr>(); in LoadInputData()
68 auto pk_node = input_nodes[i]->cast<ParameterPtr>(); in CompileGraphImpl()
94 vector<ParameterPtr> paras; in CheckModelInputs()
101 auto parameter = kernel_graph_inputs[i]->cast<ParameterPtr>(); in CheckModelInputs()
139 bool AscendInferenceSession::CompareInput(const tensor::TensorPtr &input, const ParameterPtr &param… in CompareInput()
180 std::string AscendInferenceSession::InputsInfo(const std::vector<ParameterPtr> &paras, in InputsInfo()
Dgpu_inference_session.cc42 auto pk_node = input_nodes[i]->cast<ParameterPtr>(); in LoadInputData()
68 auto pk_node = input_nodes[i]->cast<ParameterPtr>(); in CompileGraphImpl()
94 vector<ParameterPtr> paras; in CheckModelInputs()
101 auto parameter = kernel_graph_inputs[i]->cast<ParameterPtr>(); in CheckModelInputs()
139 bool GpuInferenceSession::CompareInput(const tensor::TensorPtr &input, const ParameterPtr &paramete… in CompareInput()
180 std::string GpuInferenceSession::InputsInfo(const std::vector<ParameterPtr> &paras, in InputsInfo()
/third_party/mindspore/mindspore/core/ir/
Dfunc_graph.cc86 ParameterPtr FuncGraph::add_parameter() { in add_parameter()
88 ParameterPtr p = std::make_shared<Parameter>(this_func_graph); in add_parameter()
93 void FuncGraph::add_parameter(const ParameterPtr &p) { in add_parameter()
101 ParameterPtr FuncGraph::InsertFrontParameter() { in InsertFrontParameter()
103 ParameterPtr p = std::make_shared<Parameter>(this_func_graph); in InsertFrontParameter()
108 void FuncGraph::InsertFrontParameter(const ParameterPtr &p) { in InsertFrontParameter()
116 ParameterPtr FuncGraph::AddWeightParameter(const std::string &name) { in AddWeightParameter()
118 ParameterPtr p = std::make_shared<Parameter>(this_graph); in AddWeightParameter()
575 …meter = parameters_[parameters_.size() - hyper_param_count_ - min_param_num]->cast<ParameterPtr>(); in GetVariableArgName()
584 …t auto &parameter = parameters_[parameters_.size() - hyper_param_count_ - 1]->cast<ParameterPtr>(); in GetVariableArgName()
[all …]
Dparam_info.h96 ParameterPtr parameter() { return parameter_; } in parameter()
97 void set_parameter(const ParameterPtr &parameter) { parameter_ = parameter; } in set_parameter()
117 ParameterPtr parameter_{nullptr};
Dfunc_graph.h175 ParameterPtr add_parameter() override;
176 void add_parameter(const ParameterPtr &p) final;
177 void append_parameter(const ParameterPtr &p) { parameters_.push_back(p); } in append_parameter()
179 virtual ParameterPtr InsertFrontParameter();
180 void InsertFrontParameter(const ParameterPtr &p);
181 void PrependParameter(const ParameterPtr &p) { parameters_.insert(parameters_.begin(), p); } in PrependParameter()
184 ParameterPtr AddWeightParameter(const std::string &name);
203 virtual ParameterPtr add_weight(const tensor::MetaTensorPtr &meta_tensor);
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dparameter_manager.cc85 parameter_user_info.first = parameters[0]->cast<ParameterPtr>()->name(); in FindRefKeyNodeUsers()
127 parameter_user_info.first = node->cast<ParameterPtr>()->name(); in FindParameterNodeUsers()
317 auto cloned_parameter = parameter_node->cast<ParameterPtr>(); in ParameterIsCloned()
377 static bool IsFullySplitParameter(const ParameterPtr &param_ptr) { in IsFullySplitParameter()
427 auto param_ptr = parameter->cast<ParameterPtr>(); in HandleFullySplitParameters()
458 auto cloned_parameter = cloned_parameter_node->cast<ParameterPtr>(); in SetClonedTensorShapeForOptimizer()
473 ParameterPtr cloned_from_parameter = nullptr; in SetClonedTensorShapeForOptimizer()
477 auto be_cloned_parameter = be_cloned_parameter_node->cast<ParameterPtr>(); in SetClonedTensorShapeForOptimizer()
502 std::string param_name = cloned_parameter_node->cast<ParameterPtr>()->name(); in SetClonedTensorShapeForOptimizer()
543 auto param = param_node->cast<ParameterPtr>(); in HandleAdaFactorOpt()
[all …]
/third_party/mindspore/tests/ut/cpp/ir/
Dmanager_test.cc255 ParameterPtr x = func_graph->add_parameter(); in MakeFuncGraph()
256 ParameterPtr y = func_graph->add_parameter(); in MakeFuncGraph()
280 ParameterPtr x = f->add_parameter(); in MakeNestedGraph()
308 ParameterPtr x = graph_foo->add_parameter(); in MakeNestedGraph2()
309 ParameterPtr y = graph_foo->add_parameter(); in MakeNestedGraph2()
315 ParameterPtr x1 = graph_bar->add_parameter(); in MakeNestedGraph2()
360 ParameterPtr x = func_graph->add_parameter(); in TEST_F()
361 ParameterPtr y = func_graph->add_parameter(); in TEST_F()
439 ParameterPtr x = f->add_parameter(); in TEST_F()
440 ParameterPtr y = f->add_parameter(); in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Dactor_common.cc61 …rameter_data = node->isa<Parameter>() && (!AnfAlgo::IsParameterWeight(node->cast<ParameterPtr>())); in IsHostQueueDSActor()
94 if (node->isa<Parameter>() && (!AnfAlgo::IsParameterWeight(node->cast<ParameterPtr>()))) { in IsInternalParameter()
130 if (node->isa<Parameter>() && AnfAlgo::IsParameterWeight(node->cast<ParameterPtr>())) { in IsPersistentDeviceTensor()
139 …if (front_node->isa<Parameter>() && (!AnfAlgo::IsParameterWeight(front_node->cast<ParameterPtr>())… in IsGatherActor()
/third_party/mindspore/tests/ut/cpp/session/
Dsession_basic_test.cc78 auto first_input = inputs[0]->cast<ParameterPtr>(); in TEST_F()
81 auto second_input = inputs[1]->cast<ParameterPtr>(); in TEST_F()
84 auto third_input = inputs[2]->cast<ParameterPtr>(); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dparam_replace.h43 auto param_node = node->cast<ParameterPtr>(); in operator()
49 auto para = tnode->cast<ParameterPtr>(); in operator()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dinfershape_pass.cc74 int ModifySubGraphInputCNodeFormat(const FuncGraphPtr &sub_graph, const ParameterPtr &certain_input, in ModifySubGraphInputCNodeFormat()
239 auto param_node = node->cast<ParameterPtr>(); in SetSubGraphInput()
274 if (utils::isa<ParameterPtr>(cnode->input(index))) { in SetSubGraphInput()
275 if (cnode->input(index)->cast<ParameterPtr>()->has_default()) { in SetSubGraphInput()
276 param_node->set_default_param(cnode->input(index)->cast<ParameterPtr>()->default_param()); in SetSubGraphInput()
320 if (utils::isa<ParameterPtr>(trans_input)) { in SetSubGraphOutput()
321 trans_input->cast<ParameterPtr>()->set_name(node_name); in SetSubGraphOutput()
395 auto sub_param_input = sub_input->cast<ParameterPtr>(); in ResetSubGraphInput()

12345678