sfqcyy
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!4156 fix populate parameter 

Merge pull request !4156 from sunsuodong/fix_populate_parameter
This commit is contained in:
mindspore-ci-bot 2020-08-08 16:27:18 +08:00 committed by Gitee
parent 0df5a56159 334ae54146
commit 0e0eafdacb
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1249,6 +1249,16 @@ PopulateParameterRegistry::PopulateParameterRegistry() {
populate_parameter_funcs_[schema::PrimitiveType_Add] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_Sub] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_Div] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_LogicalAnd] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_LogicalOr] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_Equal] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_Less] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_Greater] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_NotEqual] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_LessEqual] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_GreaterEqual] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_Maximum] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_Minimum] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_FloorDiv] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_FloorMod] = PopulateArithmetic;
populate_parameter_funcs_[schema::PrimitiveType_SquaredDifference] = PopulateArithmetic;
@ -1266,6 +1276,7 @@ PopulateParameterRegistry::PopulateParameterRegistry() {
populate_parameter_funcs_[schema::PrimitiveType_LogicalNot] = PopulateArithmeticSelf;
populate_parameter_funcs_[schema::PrimitiveType_Floor] = PopulateArithmeticSelf;
populate_parameter_funcs_[schema::PrimitiveType_Ceil] = PopulateArithmeticSelf;
populate_parameter_funcs_[schema::PrimitiveType_Round] = PopulateArithmeticSelf;
populate_parameter_funcs_[schema::PrimitiveType_ArgMax] = PopulateArgMaxParameter;
populate_parameter_funcs_[schema::PrimitiveType_ArgMin] = PopulateArgMinParameter;
populate_parameter_funcs_[schema::PrimitiveType_Cast] = PopulateCastParameter;