diff --git a/tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc b/tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc index 5f2ee69..3a6c6e4 100644 --- a/tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc +++ b/tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc @@ -3634,40 +3634,51 @@ TEST_F(UtestTensorflowParser, tensorflow_Message2Operator_ParseOperatorAttrs_suc tensorflow::AttrValue value; value.set_s("string"); - TensorFlowUtil::AddNodeAttr("op_def", value, &nodedef); + TensorFlowUtil::AddNodeAttr("str", value, &nodedef); + value.clear_value(); value.set_i(1); TensorFlowUtil::AddNodeAttr("num", value, &nodedef); + value.clear_value(); domi::tensorflow::DataType VALUE_TYPE = domi::tensorflow::DataType::DT_FLOAT; value.set_type(VALUE_TYPE); TensorFlowUtil::AddNodeAttr("float", value, &nodedef); + value.clear_value(); VALUE_TYPE = domi::tensorflow::DataType::DT_UINT32; value.set_type(VALUE_TYPE); TensorFlowUtil::AddNodeAttr("uint32", value, &nodedef); + value.clear_value(); VALUE_TYPE = domi::tensorflow::DataType::DT_INT64; value.set_type(VALUE_TYPE); TensorFlowUtil::AddNodeAttr("int64", value, &nodedef); + value.clear_value(); VALUE_TYPE = domi::tensorflow::DataType::DT_BOOL; value.set_type(VALUE_TYPE); TensorFlowUtil::AddNodeAttr("bool", value, &nodedef); Status ret = mess2Op.ParseOperatorAttrs(&node_def, 1, ops); EXPECT_EQ(ret, SUCCESS); + value.clear_value(); value.mutable_list()->set_s("string"); - TensorFlowUtil::AddNodeAttr("op_def", value, &nodedef); + TensorFlowUtil::AddNodeAttr("str_list", value, &nodedef); + value.clear_value(); value.mutable_list()->set_i(1); - TensorFlowUtil::AddNodeAttr("num", value, &nodedef); - domi::tensorflow::DataType VALUE_TYPE = domi::tensorflow::DataType::DT_FLOAT; + TensorFlowUtil::AddNodeAttr("num_list", value, &nodedef); + value.clear_value(); + VALUE_TYPE = domi::tensorflow::DataType::DT_FLOAT; value.mutable_list()->add_type(VALUE_TYPE); - TensorFlowUtil::AddNodeAttr("float", value, &nodedef); + TensorFlowUtil::AddNodeAttr("float_list", value, &nodedef); + value.clear_value(); VALUE_TYPE = domi::tensorflow::DataType::DT_UINT32; value.mutable_list()->add_type(VALUE_TYPE); - TensorFlowUtil::AddNodeAttr("uint32", value, &nodedef); + TensorFlowUtil::AddNodeAttr("uint32_list", value, &nodedef); + value.clear_value(); VALUE_TYPE = domi::tensorflow::DataType::DT_INT64; value.mutable_list()->add_type(VALUE_TYPE); - TensorFlowUtil::AddNodeAttr("int64", value, &nodedef); + TensorFlowUtil::AddNodeAttr("int64_list", value, &nodedef); + value.clear_value(); VALUE_TYPE = domi::tensorflow::DataType::DT_BOOL; value.mutable_list()->add_type(VALUE_TYPE); - TensorFlowUtil::AddNodeAttr("bool", value, &nodedef); + TensorFlowUtil::AddNodeAttr("bool_list", value, &nodedef); ret = mess2Op.ParseOperatorAttrs(&node_def, 1, ops); EXPECT_EQ(ret, SUCCESS); }