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