Browse Source

clean code

pull/640/head
13291271729 3 years ago
parent
commit
bb1c16298a
1 changed files with 19 additions and 8 deletions
  1. +19
    -8
      tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc

+ 19
- 8
tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc View File

@@ -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);
}


Loading…
Cancel
Save