| @@ -3654,14 +3654,15 @@ TEST_F(UtestTensorflowParser, tensorflow_Message2Operator_ParseOperatorAttrs_suc | |||||
| VALUE_TYPE = domi::tensorflow::DataType::DT_BOOL; | VALUE_TYPE = domi::tensorflow::DataType::DT_BOOL; | ||||
| value.set_type(VALUE_TYPE); | value.set_type(VALUE_TYPE); | ||||
| TensorFlowUtil::AddNodeAttr("bool", value, &nodedef); | TensorFlowUtil::AddNodeAttr("bool", value, &nodedef); | ||||
| Status ret = mess2Op.ParseOperatorAttrs(&node_def, 1, ops); | |||||
| Status ret = mess2Op.ParseOperatorAttrs(&nodedef, 1, ops); | |||||
| EXPECT_EQ(ret, SUCCESS); | EXPECT_EQ(ret, SUCCESS); | ||||
| value.clear_value(); | 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); | TensorFlowUtil::AddNodeAttr("str_list", value, &nodedef); | ||||
| value.clear_value(); | value.clear_value(); | ||||
| value.mutable_list()->set_i(1); | |||||
| value.mutable_list()->add_i(1); | |||||
| TensorFlowUtil::AddNodeAttr("num_list", value, &nodedef); | TensorFlowUtil::AddNodeAttr("num_list", value, &nodedef); | ||||
| value.clear_value(); | value.clear_value(); | ||||
| VALUE_TYPE = domi::tensorflow::DataType::DT_FLOAT; | 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_TYPE = domi::tensorflow::DataType::DT_BOOL; | ||||
| value.mutable_list()->add_type(VALUE_TYPE); | value.mutable_list()->add_type(VALUE_TYPE); | ||||
| TensorFlowUtil::AddNodeAttr("bool_list", value, &nodedef); | TensorFlowUtil::AddNodeAttr("bool_list", value, &nodedef); | ||||
| ret = mess2Op.ParseOperatorAttrs(&node_def, 1, ops); | |||||
| ret = mess2Op.ParseOperatorAttrs(&nodedef, 1, ops); | |||||
| EXPECT_EQ(ret, SUCCESS); | EXPECT_EQ(ret, SUCCESS); | ||||
| } | } | ||||