Home
last modified time | relevance | path

Searched refs:act_param (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/lite/src/ops/populate/v0/
Dactivation_populate_v0.cc32 auto *act_param = reinterpret_cast<ActivationParameter *>(malloc(sizeof(ActivationParameter))); in PopulateActivationParameter() local
33 if (act_param == nullptr) { in PopulateActivationParameter()
37 memset(act_param, 0, sizeof(ActivationParameter)); in PopulateActivationParameter()
38 act_param->op_parameter_.type_ = schema::PrimitiveType_Activation; in PopulateActivationParameter()
40 act_param->type_ = static_cast<int>(activation_prim->type()); in PopulateActivationParameter()
41 act_param->alpha_ = activation_prim->alpha(); in PopulateActivationParameter()
42 act_param->min_val_ = activation_prim->min_val(); in PopulateActivationParameter()
43 act_param->max_val_ = activation_prim->max_val(); in PopulateActivationParameter()
44 return reinterpret_cast<OpParameter *>(act_param); in PopulateActivationParameter()
Dactivation_grad_populate_v0.cc32 …auto *act_param = reinterpret_cast<ActivationGradParameter *>(malloc(sizeof(ActivationGradParamete… in PopulateActivationGradParameter() local
33 if (act_param == nullptr) { in PopulateActivationGradParameter()
37 memset(act_param, 0, sizeof(ActivationGradParameter)); in PopulateActivationGradParameter()
38 act_param->op_parameter.type_ = schema::PrimitiveType_ActivationGrad; in PopulateActivationGradParameter()
40 act_param->type_ = static_cast<int>(activation_grad_prim->type()); in PopulateActivationGradParameter()
41 act_param->alpha_ = activation_grad_prim->alpha(); in PopulateActivationGradParameter()
42 return reinterpret_cast<OpParameter *>(act_param); in PopulateActivationGradParameter()
Dgelu_populate_v0.cc32 auto *act_param = reinterpret_cast<ActivationParameter *>(malloc(sizeof(ActivationParameter))); in PopulateGeLUParameter() local
33 if (act_param == nullptr) { in PopulateGeLUParameter()
37 memset(act_param, 0, sizeof(ActivationParameter)); in PopulateGeLUParameter()
38 act_param->op_parameter_.type_ = schema::PrimitiveType_Activation; in PopulateGeLUParameter()
39 act_param->type_ = static_cast<int>(schema::ActivationType_GELU); in PopulateGeLUParameter()
40 return reinterpret_cast<OpParameter *>(act_param); in PopulateGeLUParameter()
Dclip_populate_v0.cc24 OpParameter *act_param = reinterpret_cast<OpParameter *>(malloc(sizeof(OpParameter))); in PopulateClipParameter() local
25 if (act_param == nullptr) { in PopulateClipParameter()
29 memset(act_param, 0, sizeof(OpParameter)); in PopulateClipParameter()
30 act_param->type_ = schema::PrimitiveType_Clip; in PopulateClipParameter()
31 return reinterpret_cast<OpParameter *>(act_param); in PopulateClipParameter()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/
Dopencl_fusion.cc330 auto *act_param = in TryMergeArithmeticAct() local
333 MS_ASSERT(act_param); in TryMergeArithmeticAct()
336 (act_param->type_ == ActivationType_RELU || act_param->type_ == ActivationType_RELU6)) { in TryMergeArithmeticAct()
337 arithmetic_param->activation_type_ = act_param->type_; in TryMergeArithmeticAct()
349 auto *act_param = in TryMergeXxxActivation() local
370 if (act_param->type_ == ActivationType_RELU) { in TryMergeXxxActivation()
372 } else if (act_param->type_ == ActivationType_RELU6) { in TryMergeXxxActivation()
374 } else if (act_param->type_ == ActivationType_TANH) { in TryMergeXxxActivation()
381 param->act_type_ = static_cast<ActType>(act_param->type_); in TryMergeXxxActivation()
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_populate_parameter.cc266 …ActivationParameter *act_param = reinterpret_cast<ActivationParameter *>(malloc(sizeof(ActivationP… in PopulateActivationGradParameter() local
267 if (act_param == nullptr) { in PopulateActivationGradParameter()
271 memset(act_param, 0, sizeof(ActivationParameter)); in PopulateActivationGradParameter()
275 act_param->op_parameter_.type_ = primitive->value_type(); in PopulateActivationGradParameter()
276 act_param->type_ = static_cast<int>(value->activation_type()); in PopulateActivationGradParameter()
277 act_param->alpha_ = value->alpha(); in PopulateActivationGradParameter()
278 return reinterpret_cast<OpParameter *>(act_param); in PopulateActivationGradParameter()
Dtrain_populate_parameter_v0.cc327 …ActivationParameter *act_param = reinterpret_cast<ActivationParameter *>(malloc(sizeof(ActivationP… in PopulateActivationGradParameter() local
328 if (act_param == nullptr) { in PopulateActivationGradParameter()
332 memset(act_param, 0, sizeof(ActivationParameter)); in PopulateActivationGradParameter()
333 act_param->op_parameter_.type_ = schema::PrimitiveType_ActivationGrad; in PopulateActivationGradParameter()
336 act_param->type_ = static_cast<int>(activationGrad_prim->type()); in PopulateActivationGradParameter()
337 act_param->alpha_ = activationGrad_prim->alpha(); in PopulateActivationGradParameter()
338 return reinterpret_cast<OpParameter *>(act_param); in PopulateActivationGradParameter()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dactivation_fp16.cc123 auto act_param = reinterpret_cast<ActivationParameter *>(opParameter); in CpuActivationFp16KernelCreator() local
124 auto type = act_param->type_; in CpuActivationFp16KernelCreator()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/
Druntime_pass_tests.cc173 OpParameter *act_param = new OpParameter(); in ConvNormC4PassConstruct() local
174 act_param->type_ = schema::PrimitiveType_Activation; in ConvNormC4PassConstruct()
177 reg->GetKernel(act_in, act_out, ctx, nullptr, act_desc, act_param, &act_kernel, nullptr); in ConvNormC4PassConstruct()