p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!4136 fix bias add populate paremeter 

Merge pull request !4136 from wangchangkai/master
This commit is contained in:
mindspore-ci-bot 2020-08-10 09:13:27 +08:00 committed by Gitee
parent d92eb62852 cb95589f0d
commit 3da5d72419
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mindspore/lite/src/populate_parameter.cc
View File

@ -1228,6 +1228,17 @@ OpParameter *PopulateEmbeddingLookupParameter(const lite::Primitive *primitive)
return reinterpret_cast<OpParameter *>(embedding_lookup_parameter);
}
OpParameter *PopulateBiasAddParameter(const lite::Primitive *primitive) {
ArithmeticParameter *arithmetic_param = new (std::nothrow) ArithmeticParameter();
if (arithmetic_param == nullptr) {
MS_LOG(ERROR) << "new Bias Add Parameter failed";
return nullptr;
}
arithmetic_param->op_parameter_.type_ = primitive->Type();
return reinterpret_cast<OpParameter *>(arithmetic_param);
}
PopulateParameterRegistry::PopulateParameterRegistry() {
populate_parameter_funcs_[schema::PrimitiveType_SoftMax] = PopulateSoftmaxParameter;
populate_parameter_funcs_[schema::PrimitiveType_Activation] = PopulateActivationParameter;
@ -1262,7 +1273,7 @@ PopulateParameterRegistry::PopulateParameterRegistry() {
populate_parameter_funcs_[schema::PrimitiveType_FloorDiv] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_FloorMod] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_SquaredDifference] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_BiasAdd] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_BiasAdd] = PopulateBiasAddParameter;
populate_parameter_funcs_[schema::PrimitiveType_Eltwise] = PopulateEltwiseParameter;
populate_parameter_funcs_[schema::PrimitiveType_ExpandDims] = PopulateExpandDimsParameter;
populate_parameter_funcs_[schema::PrimitiveType_Abs] = PopulateArithmeticSelf;