diff --git a/mindspore/lite/src/populate_parameter.cc b/mindspore/lite/src/populate_parameter.cc index 6a757c8041..dba8e3e331 100644 --- a/mindspore/lite/src/populate_parameter.cc +++ b/mindspore/lite/src/populate_parameter.cc @@ -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;