提交 1a860698 编写于 作者: G guru4elephant

add int_data and float_data for Tensor

上级 624e0fb3
...@@ -21,8 +21,10 @@ option cc_generic_services = true; ...@@ -21,8 +21,10 @@ option cc_generic_services = true;
message Tensor { message Tensor {
repeated bytes data = 1; repeated bytes data = 1;
optional int32 elem_type = 2; repeated int32 int_data = 2;
repeated int32 shape = 3; repeated float float_data = 3;
optional int32 elem_type = 4;
repeated int32 shape = 5;
}; };
message FeedInst { message FeedInst {
......
...@@ -20,9 +20,11 @@ package baidu.paddle_serving.predictor.general_model; ...@@ -20,9 +20,11 @@ package baidu.paddle_serving.predictor.general_model;
option cc_generic_services = true; option cc_generic_services = true;
message Tensor { message Tensor {
repeated bytes data = 1; repeated bytes data = 1; // most general format
optional int32 elem_type = 2; repeated int32 int_data = 2; // for simple debug only
repeated int32 shape = 3; repeated float float_data = 3; // for simple debug only
optional int32 elem_type = 4; // support int64, float32
repeated int32 shape = 5;
}; };
message FeedInst { message FeedInst {
......
...@@ -20,7 +20,9 @@ import google.protobuf.text_format ...@@ -20,7 +20,9 @@ import google.protobuf.text_format
class OpMaker(object): class OpMaker(object):
def __init__(self): def __init__(self):
self.op_dict = {"general_infer":"GeneralInferOp", self.op_dict = {"general_infer":"GeneralInferOp",
"general_text_infer":"GeneralTextInferOp",
"general_reader":"GeneralReaderOp", "general_reader":"GeneralReaderOp",
"general_text_reader":"GeneralTextReaderOp",
"general_single_kv":"GeneralSingleKVOp", "general_single_kv":"GeneralSingleKVOp",
"general_dist_kv":"GeneralDistKVOp"} "general_dist_kv":"GeneralDistKVOp"}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册