Home
last modified time | relevance | path

Searched refs:default_param (Results 1 – 25 of 54) sorted by relevance

123

/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
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()
Dconv_conv_fusion.cc64 … auto down_bias_param = std::dynamic_pointer_cast<tensor::Tensor>(down_bias_node->default_param()); in GenNewConvBias()
74 auto up_bias_param = std::dynamic_pointer_cast<tensor::Tensor>(up_bias_node->default_param()); in GenNewConvBias()
81 …o down_weight_param = std::dynamic_pointer_cast<tensor::Tensor>(down_weight_node->default_param()); in GenNewConvBias()
120 …o down_weight_param = std::dynamic_pointer_cast<tensor::Tensor>(down_weight_node->default_param()); in GenNewConvWeight()
124 auto up_weight_param = std::dynamic_pointer_cast<tensor::Tensor>(up_weight_node->default_param()); in GenNewConvWeight()
251 …n_weight_value = std::dynamic_pointer_cast<tensor::Tensor>(down_weight_parameter->default_param()); in CheckCanFusion()
265 … up_weight_value = std::dynamic_pointer_cast<tensor::Tensor>(up_weight_parameter->default_param()); in CheckCanFusion()
Dgelu_fusion.cc58 if (!parameter_node->has_default() || parameter_node->default_param() == nullptr) { in GetParameterValue()
61 auto param_value_lite = parameter_node->default_param()->cast<tensor::TensorPtr>(); in GetParameterValue()
Dconv_scale_fusion.cc68 auto scale_weight_param = scale_weight_node->cast<ParameterPtr>()->default_param(); in InitTransParam()
81 auto scale_bias_param = scale_bias_node->cast<ParameterPtr>()->default_param(); in InitTransParam()
Dtf_lstm_cell_fusion.cc233 if (!weight_param->has_default() || weight_param->default_param() == nullptr) { in SplitWeights()
237 if (!utils::isa<tensor::TensorPtr>(weight_param->default_param())) { in SplitWeights()
241 auto origin_tensor = std::dynamic_pointer_cast<tensor::Tensor>(weight_param->default_param()); in SplitWeights()
286 if (!old_bias_param->has_default() || old_bias_param->default_param() == nullptr) { in PopulateBiasNode()
290 if (!utils::isa<tensor::TensorPtr>(old_bias_param->default_param())) { in PopulateBiasNode()
294 auto origin_tensor = std::dynamic_pointer_cast<tensor::Tensor>(old_bias_param->default_param()); in PopulateBiasNode()
Dconv_pad_fusion.cc47 MS_ASSERT(paddings->default_param() != nullptr); in ReplaceParamsAndNodes()
48 auto pad_list = std::dynamic_pointer_cast<tensor::Tensor>(paddings->default_param()); in ReplaceParamsAndNodes()
119 auto tensor_param = pad_list->default_param(); in IsPrimitiveProper()
Dbatchmatmul_fusion.cc42 MS_CHECK_TRUE_RET(param_input->default_param() != nullptr, nullptr); in GetInputAddr()
43 auto tensor_info = param_input->default_param()->cast<tensor::TensorPtr>(); in GetInputAddr()
61 auto fc_weight_param = std::dynamic_pointer_cast<tensor::Tensor>(fc_weight->default_param()); in GetRightMatmulInputParamter()
Dtflite_lstm_cell_fusion.cc127 if (!param_ptr->has_default() || param_ptr->default_param() == nullptr) { in GetFloatScalarFromTensorInfo()
131 auto default_param = param_ptr->default_param(); in GetFloatScalarFromTensorInfo() local
132 if (!utils::isa<tensor::TensorPtr>(default_param)) { in GetFloatScalarFromTensorInfo()
136 auto default_param_ptr = utils::cast<tensor::TensorPtr>(default_param); in GetFloatScalarFromTensorInfo()
432 if (!param_t->has_default() || param_t->default_param() == nullptr) { in GetConcatedParam()
436 if (!utils::isa<tensor::TensorPtr>(param_t->default_param())) { in GetConcatedParam()
440 auto origin_tensor = std::dynamic_pointer_cast<tensor::Tensor>(param_t->default_param()); in GetConcatedParam()
Dmulti_head_attention_fusion.cc294 auto default_param = param_ptr->default_param(); in GetIntParameterData() local
295 if (default_param == nullptr || !utils::isa<tensor::TensorPtr>(default_param)) { in GetIntParameterData()
299 auto default_param_ptr = utils::cast<tensor::TensorPtr>(default_param); in GetIntParameterData()
Dnorm_fusion.cc35 if (!axes_param->has_default() || axes_param->default_param() == nullptr) { in GetReduceAxes()
38 auto axes_value = axes_param->default_param()->cast<tensor::TensorPtr>(); in GetReduceAxes()
213 auto beta_param = beta_node->cast<ParameterPtr>()->default_param(); in CheckPattern()
226 auto gamma_param = gamma_node->cast<ParameterPtr>()->default_param(); in CheckPattern()
239 auto epsilon_param = epsilon_node->cast<ParameterPtr>()->default_param(); in CheckPattern()
Dconv_transform_fusion.cc201 auto conv_weight_param = conv_weight_node->cast<ParameterPtr>()->default_param(); in GenNewConvTensor()
222 auto conv_bias_param = conv_bias_node->cast<ParameterPtr>()->default_param(); in GenNewConvTensor()
364 …conv_weight_node->isa<Parameter>() ? conv_weight_node->cast<ParameterPtr>()->default_param() : nul… in IsVariableWeightConv()
Dmatmul_add_fusion.cc72 (!utils::isa<Parameter>(bias_node) || !bias_node->cast<ParameterPtr>()->default_param())) { in Run()
Dtflite_rel_pos_multi_head_attention_fusion.cc161 auto default_param = param_ptr->default_param(); in GetIntParameterData() local
162 if (!utils::isa<tensor::TensorPtr>(default_param)) { in GetIntParameterData()
166 auto default_param_ptr = utils::cast<tensor::TensorPtr>(default_param); in GetIntParameterData()
/third_party/mindspore/mindspore/lite/tools/converter/acl/common/
Dutils.cc135 auto default_param = param_ptr->default_param(); in GetIntParameterData() local
136 if (!utils::isa<tensor::TensorPtr>(default_param)) { in GetIntParameterData()
140 auto default_param_ptr = utils::cast<tensor::TensorPtr>(default_param); in GetIntParameterData()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dunused_transpose_node_remove_pass.cc46 if (!perm_param->has_default() || perm_param->default_param() == nullptr) { in GetTransposePerm()
49 auto perm_value = perm_param->default_param()->cast<tensor::TensorPtr>(); in GetTransposePerm()
Dslice_prepose_pass.cc49 …if (param_node == nullptr || !param_node->has_default() || param_node->default_param() == nullptr)… in GetSliceBeginAndSize()
52 auto tensor_info = param_node->default_param()->cast<tensor::TensorPtr>(); in GetSliceBeginAndSize()
89 auto default_param = param->default_param(); in GetDefaultParamShape() local
90 if (default_param == nullptr) { in GetDefaultParamShape()
94 if (!utils::isa<tensor::TensorPtr>(default_param)) { in GetDefaultParamShape()
98 auto param_value_lite = utils::cast<tensor::TensorPtr>(default_param); in GetDefaultParamShape()
109 … auto tensor = utils::cast<tensor::TensorPtr>(utils::cast<ParameterPtr>(nodePtr)->default_param()); in IsScalarNode()
162 if (!perm_param->has_default() || perm_param->default_param() == nullptr) { in GetTransposePerm()
165 auto perm_value = perm_param->default_param()->cast<tensor::TensorPtr>(); in GetTransposePerm()
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_inputs_adjust.cc95 auto default_value = param_node->default_param(); in ReplaceInt64ParameterNode()
121 auto tensor_info = std::dynamic_pointer_cast<tensor::Tensor>(param_node->default_param()); in ValidParameterNode()
201 !utils::isa<tensor::TensorPtr>(perm_param->default_param())) { in ReplaceTransposeWithGraphInput()
205 auto perm_value = perm_param->default_param()->cast<tensor::TensorPtr>(); in ReplaceTransposeWithGraphInput()
256 const auto &default_data = param_node->default_param()->cast<tensor::TensorPtr>(); in AdjustStridedSlice()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Ddynamic_assign_cpu_kernel.cc97 auto value = node_ptr->default_param(); in LaunchKernel()
/third_party/mindspore/mindspore/ccsrc/utils/
Dcallbacks_ge.cc56 tensor = std::dynamic_pointer_cast<tensor::Tensor>(param_node->default_param()); in GetParameterShape()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dkpynative.cc999 p->set_default_param(input_w->default_param()); in SetSensAndWeights()
1036 auto default_param = input_w->default_param(); in SetOutput() local
1037 MS_EXCEPTION_IF_NULL(default_param); in SetOutput()
1038 auto dout = BuildZerosLikeValue(tape_, default_param); in SetOutput()
/third_party/mindspore/mindspore/ccsrc/fl/server/
Dcommon.h247 auto param_tensor = param_node->default_param()->cast<tensor::TensorPtr>(); in GenerateParameterNodeAddrPtr()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_inputs_adjust.cc115 auto default_value = param_node->default_param(); in ReplaceInt64ParameterNode()
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/
Dmodel_converter.cc35 auto value = para->default_param(); in GetParams()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_inference_session.cc73 const auto &param_value = pk_node->default_param(); in CompileGraphImpl()
Dgpu_inference_session.cc73 const auto &param_value = pk_node->default_param(); in CompileGraphImpl()

123