Home
last modified time | relevance | path

Searched defs:param_node (Results 1 – 25 of 33) sorted by relevance

12

/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_inputs_adjust.cc55 auto param_node = in AddAttrToInput() local
72 STATUS ReplaceInt64ParameterNode(const FuncGraphPtr &func_graph, const ParameterPtr &param_node) { in ReplaceInt64ParameterNode()
116 bool ValidParameterNode(const ParameterPtr &param_node) { in ValidParameterNode()
158 auto param_node = opt::BuildParameterNode(func_graph, cnode, tensor_info_ptr); in ReplaceConstant() local
178 auto param_node = anf_node->cast<ParameterPtr>(); in ReplaceTransposeWithGraphInput() local
252 const auto &param_node = cnode->input(opt::kInputIndexTwo)->cast<ParameterPtr>(); in AdjustStridedSlice() local
334 auto param_node = in AdjustResize() local
373 auto param_node = node->cast<ParameterPtr>(); in Adjust() local
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dweight_quantizer.cc61 … WeightQuantizer::SetAbstract(const tensor::TensorPtr &tensor_info, const ParameterPtr &param_node, in SetAbstract()
100 ParameterPtr param_node; in DoConvQuantize() local
144 auto param_node = inputNode->cast<ParameterPtr>(); in DoMulQuantize() local
228 ParameterPtr param_node; in DoGatherQuantize() local
286 ParameterPtr param_node; in DoOptimizerQuantize() local
332 ParameterPtr param_node; in DoMarkWeightQuantizeIfQuantized() local
351 ParameterPtr param_node; in ProcessLstmWeightByIndex() local
Dquantize_util.cc604 auto param_node = input_node->cast<ParameterPtr>(); in CopyFuncGraph() local
631 void GetLiteParameter(const AnfNodePtr &node, ParameterPtr *param_node, tensor::TensorPtr *tensor_i… in GetLiteParameter()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dparam_replace.h43 auto param_node = node->cast<ParameterPtr>(); in operator() local
/third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/
Dprimitive_mapper.cc180 auto param_node = in AddAttrToInput() local
187 auto param_node = in AddAttrToInput() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/
Dpull_kernel.h62 auto param_node = AnfAlgo::GetInputNode(kernel_node, 1); in Init() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcontext.cc232 … const ParameterPtr &param_node, const AbstractBasePtr &ptr) { in ParallelParameterContextRestoreShape()
256 …::ParallelParameterContextCkptShape(const FuncGraphPtr &func_graph, const ParameterPtr &param_node, in ParallelParameterContextCkptShape()
Dparameter_manager.cc541 for (auto &param_node : root->parameters()) { in HandleAdaFactorOpt() local
/third_party/mindspore/mindspore/lite/tools/anf_exporter/
Dfetch_content.cc90 STATUS GetDataTypeAndShape(const ParameterPtr &param_node, TypeId *data_type, ShapeVector *shape_ve… in GetDataTypeAndShape()
234 int FetchFromDefaultParam(const ParameterPtr &param_node, const converter::FmkType &fmk_type, DataI… in FetchFromDefaultParam()
273 auto param_node = cnode->input(index)->cast<ParameterPtr>(); in FetchDataFromParameterNode() local
Danf_exporter.cc229 auto param_node = input->cast<ParameterPtr>(); in CreateNewTensorForParameter() local
684 auto param_node = cnode->input(index)->cast<ParameterPtr>(); in ConvertInputParameter() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dinputs_adjust.cc53 AnfNodePtr param_node = nullptr; in AddAttrToInput() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dinfershape_pass.cc239 auto param_node = node->cast<ParameterPtr>(); in SetSubGraphInput() local
390 auto param_node = sub_graph->add_parameter(); in ResetSubGraphInput() local
Dtranspose_strategy.cc121 …auto param_node = BuildIntVecParameterNode(func_graph, cur_input, cnode->input(input_index)->fulln… in TransformAttrByAxes() local
244 auto param_node = in ChangeOpPad() local
332 auto param_node = in ChangeOpStrideSlice() local
Ddecrease_transpose_algo.cc190 auto param_node = func_graph->add_parameter(); in ConvertTensorToNCOrNH() local
481 auto param_node = node->cast<ParameterPtr>(); in SetSubGraphInput() local
536 auto param_node = sub_graph->add_parameter(); in ResetSubGraphInput() local
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc897 auto param_node = func_graph->add_parameter(); in BuildParameterNode() local
958 auto param_node = func_graph->add_parameter(); in BuildIntValueParameterNode() local
979 auto param_node = func_graph->add_parameter(); in BuildIntVecParameterNode() local
1002 auto param_node = func_graph->add_parameter(); in BuildIntVec2DParameterNode() local
1032 auto param_node = func_graph->add_parameter(); in BuildFloatValueParameterNode() local
1052 auto param_node = func_graph->add_parameter(); in BuildFloatVecParameterNode() local
/third_party/mindspore/mindspore/core/ir/
Dfunc_graph_extends.cc132 AnfNodePtr param_node = specialized_graph->GetParameterByName(kw_param_name); in GenerateKwParams() local
210 auto param_node = specialized_graph->parameters()[i]->cast<ParameterPtr>(); in GenerateDefaultValue() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_inputs_adjust.cc92 …Adjust::ReplaceInt64ParameterNode(const FuncGraphPtr &func_graph, const ParameterPtr &param_node) { in ReplaceInt64ParameterNode()
/third_party/mindspore/mindspore/ccsrc/utils/
Dcallbacks_ge.cc48 ParameterPtr param_node = std::static_pointer_cast<Parameter>(node); in GetParameterShape() local
/third_party/mindspore/mindspore/lite/tools/common/
Dtensor_util.cc181 int InitParameterFromTensorInfo(const ParameterPtr &param_node, const tensor::TensorPtr &tensor_inf… in InitParameterFromTensorInfo()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Daction.cc64 auto param_node = param->cast<ParameterPtr>(); in UpdateFuncGraphParameter() local
515 auto param_node = std::static_pointer_cast<Parameter>(param); in AbstractSpecializeAction() local
1000 auto param_node = std::static_pointer_cast<Parameter>(param); in SetMindIRGraphAction() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dnode_info.cc252 for (auto &param_node : root_g->parameters()) { in FindParameterByRefKeyNode() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/allreduce_fusion/
Dallreduce_fusion.cc255 for (auto &param_node : paras) { in FindMirrorAndSetFusion() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dpy_pass.cc224 auto param_node = param->cast<ParameterPtr>(); in ReflectParamBackToPython() local
/third_party/mindspore/mindspore/ccsrc/fl/server/
Dcommon.h244 auto param_node = in GenerateParameterNodeAddrPtr() local
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dresolve.cc115 auto param_node = dyn_cast<Parameter>(param); in ResolveParameterObj() local

12