提交 f3d7bb84 编写于 作者: M MRXLT

fix grpc response for tensor

上级 876ebaec
...@@ -584,7 +584,7 @@ class MultiLangServerServiceServicer(multi_lang_general_model_service_pb2_grpc. ...@@ -584,7 +584,7 @@ class MultiLangServerServiceServicer(multi_lang_general_model_service_pb2_grpc.
else: else:
raise Exception("error type.") raise Exception("error type.")
tensor.shape.extend(list(model_result[name].shape)) tensor.shape.extend(list(model_result[name].shape))
if name in self.lod_tensor_set_: if model_result.has_key("{}.lod".format(name)):
tensor.lod.extend(model_result["{}.lod".format(name)] tensor.lod.extend(model_result["{}.lod".format(name)]
.tolist()) .tolist())
inst.tensor_array.append(tensor) inst.tensor_array.append(tensor)
......
...@@ -645,7 +645,7 @@ class MultiLangServerServiceServicer(multi_lang_general_model_service_pb2_grpc. ...@@ -645,7 +645,7 @@ class MultiLangServerServiceServicer(multi_lang_general_model_service_pb2_grpc.
else: else:
raise Exception("error type.") raise Exception("error type.")
tensor.shape.extend(list(model_result[name].shape)) tensor.shape.extend(list(model_result[name].shape))
if name in self.lod_tensor_set_: if model_result.has_key("{}.lod".format(name)):
tensor.lod.extend(model_result["{}.lod".format(name)] tensor.lod.extend(model_result["{}.lod".format(name)]
.tolist()) .tolist())
inst.tensor_array.append(tensor) inst.tensor_array.append(tensor)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册