Skip to content

M
mindspore

magicwindyyd / mindspore
与 Fork 源项目一致

Fork自 MindSpore / mindspore

通知 1
Star 1
Fork 0
M
mindspore
提交 0e0eafda 编写于 作者: M mindspore-ci-bot 提交者： Gitee
浏览文件
操作

!4156 fix populate parameter 

Merge pull request !4156 from sunsuodong/fix_populate_parameter
上级 0df5a561 334ae541
隐藏空白更改
内联 并排
Showing with 11 addition and 0 deletion
mindspore/lite/src/populate_parameter.cc
浏览文件 @ 0e0eafda
......@@ -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;
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或