From 334ae54146835fb1dcfe5b3f8589287a5f9c9fe2 Mon Sep 17 00:00:00 2001
From: sunsuodong <sunsuodong@huawei.com>
Date: Sat, 8 Aug 2020 16:06:35 +0800
Subject: [PATCH] fix_populate_parameter

---
 mindspore/lite/src/populate_parameter.cc | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/mindspore/lite/src/populate_parameter.cc b/mindspore/lite/src/populate_parameter.cc
index 22e5b42deb..15e7622c1b 100644
--- a/mindspore/lite/src/populate_parameter.cc
+++ b/mindspore/lite/src/populate_parameter.cc
@@ -1248,6 +1248,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;
@@ -1265,6 +1275,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;