提交 5080f33b 编写于 作者: S ShiningZhang

add comment for client

上级 e8f31562
...@@ -37,6 +37,7 @@ class ServingBrpcClient : public ServingClient { ...@@ -37,6 +37,7 @@ class ServingBrpcClient : public ServingClient {
const uint64_t log_id); const uint64_t log_id);
private: private:
// generate default SDKConf
std::string gen_desc(const std::string server_port); std::string gen_desc(const std::string server_port);
private: private:
......
...@@ -143,6 +143,7 @@ class PredictorData { ...@@ -143,6 +143,7 @@ class PredictorData {
std::string print(); std::string print();
private: private:
// used to print vector data map e.g. _float_data_map
template<typename T1, typename T2> template<typename T1, typename T2>
std::string map2string(const std::map<T1, std::vector<T2>>& map) { std::string map2string(const std::map<T1, std::vector<T2>>& map) {
std::ostringstream oss; std::ostringstream oss;
...@@ -171,6 +172,7 @@ class PredictorData { ...@@ -171,6 +172,7 @@ class PredictorData {
return oss.str(); return oss.str();
}; };
// used to print data map without vector e.g. _string_data_map
template<typename T1, typename T2> template<typename T1, typename T2>
std::string map2string(const std::map<T1, T2>& map) { std::string map2string(const std::map<T1, T2>& map) {
std::ostringstream oss; std::ostringstream oss;
...@@ -203,6 +205,9 @@ class PredictorInputs : public PredictorData { ...@@ -203,6 +205,9 @@ class PredictorInputs : public PredictorData {
PredictorInputs() {}; PredictorInputs() {};
virtual ~PredictorInputs() {}; virtual ~PredictorInputs() {};
// generate proto from inputs
// feed_name_to_idx: mapping alias name to idx
// feed_name: mapping idx to name
static int GenProto(const PredictorInputs& inputs, static int GenProto(const PredictorInputs& inputs,
const std::map<std::string, int>& feed_name_to_idx, const std::map<std::string, int>& feed_name_to_idx,
const std::vector<std::string>& feed_name, const std::vector<std::string>& feed_name,
...@@ -235,6 +240,9 @@ class PredictorOutputs { ...@@ -235,6 +240,9 @@ class PredictorOutputs {
void clear(); void clear();
// Parse proto to outputs
// fetch_name: name of data to be output
// fetch_name_to_type: mapping of fetch_name to datatype
static int ParseProto(const predictor::general_model::Response& res, static int ParseProto(const predictor::general_model::Response& res,
const std::vector<std::string>& fetch_name, const std::vector<std::string>& fetch_name,
std::map<std::string, int>& fetch_name_to_type, std::map<std::string, int>& fetch_name_to_type,
......
...@@ -386,7 +386,7 @@ int PredictorOutputs::ParseProto(const Response& res, ...@@ -386,7 +386,7 @@ int PredictorOutputs::ParseProto(const Response& res,
for (auto &name : fetch_name) { for (auto &name : fetch_name) {
// int idx = _fetch_name_to_idx[name]; // int idx = _fetch_name_to_idx[name];
if (fetch_name_to_type[name] == P_INT64) { if (fetch_name_to_type[name] == P_INT64) {
VLOG(2) << "ferch var " << name << "type int64"; VLOG(2) << "fetch var " << name << "type int64";
int size = output.tensor(idx).int64_data_size(); int size = output.tensor(idx).int64_data_size();
int64_data_map[name] = std::vector<int64_t>( int64_data_map[name] = std::vector<int64_t>(
output.tensor(idx).int64_data().begin(), output.tensor(idx).int64_data().begin(),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册