Browse Source

clean code

pull/640/head
13291271729 3 years ago
parent
commit
388ba4b7d8
1 changed files with 5 additions and 4 deletions
  1. +5
    -4
      tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc

+ 5
- 4
tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc View File

@@ -3654,14 +3654,15 @@ TEST_F(UtestTensorflowParser, tensorflow_Message2Operator_ParseOperatorAttrs_suc
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);
Status ret = mess2Op.ParseOperatorAttrs(&nodedef, 1, ops);
EXPECT_EQ(ret, SUCCESS);

value.clear_value();
value.mutable_list()->set_s("string");
VALUE_TYPE = domi::tensorflow::DataType::DT_STRING;
value.mutable_list()->add_type(VALUE_TYPE);
TensorFlowUtil::AddNodeAttr("str_list", value, &nodedef);
value.clear_value();
value.mutable_list()->set_i(1);
value.mutable_list()->add_i(1);
TensorFlowUtil::AddNodeAttr("num_list", value, &nodedef);
value.clear_value();
VALUE_TYPE = domi::tensorflow::DataType::DT_FLOAT;
@@ -3679,7 +3680,7 @@ TEST_F(UtestTensorflowParser, tensorflow_Message2Operator_ParseOperatorAttrs_suc
VALUE_TYPE = domi::tensorflow::DataType::DT_BOOL;
value.mutable_list()->add_type(VALUE_TYPE);
TensorFlowUtil::AddNodeAttr("bool_list", value, &nodedef);
ret = mess2Op.ParseOperatorAttrs(&node_def, 1, ops);
ret = mess2Op.ParseOperatorAttrs(&nodedef, 1, ops);
EXPECT_EQ(ret, SUCCESS);
}



Loading…
Cancel
Save