提交 8e04aaa6 编写于 作者: X xulongteng

fix proto

上级 a5e2e351
...@@ -25,6 +25,8 @@ ...@@ -25,6 +25,8 @@
#endif #endif
#include "demo-serving/bert_service.pb.h" #include "demo-serving/bert_service.pb.h"
#include <sys/time.h>
namespace baidu { namespace baidu {
namespace paddle_serving { namespace paddle_serving {
namespace serving { namespace serving {
......
...@@ -25,8 +25,8 @@ message BertReqInstance { ...@@ -25,8 +25,8 @@ message BertReqInstance {
repeated int64 sentence_type_ids = 2; repeated int64 sentence_type_ids = 2;
repeated int64 position_ids = 3; repeated int64 position_ids = 3;
repeated float input_masks = 4; repeated float input_masks = 4;
required int64 max_seq_len = 5; optional int64 max_seq_len = 5;
required int64 emb_size = 6; optional int64 emb_size = 6;
}; };
message Request { repeated BertReqInstance instances = 1; }; message Request { repeated BertReqInstance instances = 1; };
......
...@@ -25,7 +25,8 @@ message BertReqInstance { ...@@ -25,7 +25,8 @@ message BertReqInstance {
repeated int64 sentence_type_ids = 2; repeated int64 sentence_type_ids = 2;
repeated int64 position_ids = 3; repeated int64 position_ids = 3;
repeated float input_masks = 4; repeated float input_masks = 4;
required int64 max_seq_len = 5; optional int64 max_seq_len = 5;
optional int64 emb_size = 6;
}; };
message Request { repeated BertReqInstance instances = 1; }; message Request { repeated BertReqInstance instances = 1; };
...@@ -34,7 +35,11 @@ message Embedding_values { repeated float values = 1; }; ...@@ -34,7 +35,11 @@ message Embedding_values { repeated float values = 1; };
message BertResInstance { repeated Embedding_values instances = 1; }; message BertResInstance { repeated Embedding_values instances = 1; };
message Response { repeated BertResInstance instances = 1; }; message Response {
repeated BertResInstance instances = 1;
optional int64 op_time = 2;
optional int64 infert_time = 3;
};
service BertService { service BertService {
rpc inference(Request) returns (Response); rpc inference(Request) returns (Response);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册