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