未验证 提交 196c9f96 编写于 作者: X Xin Pan 提交者: GitHub

Merge pull request #13192 from seiriosPlus/cherry-pick-selected_rows_size_fix

Cherry-pick-SelectedRows-Size-Fix
...@@ -1060,7 +1060,8 @@ class DistributeTranspiler(object): ...@@ -1060,7 +1060,8 @@ class DistributeTranspiler(object):
self.table_name] self.table_name]
zero_dim = int( zero_dim = int(
math.ceil(origin_param_var.shape[0] / len(self.pserver_endpoints))) math.ceil(origin_param_var.shape[0] / float(
len(self.pserver_endpoints))))
table_shape = list(origin_param_var.shape) table_shape = list(origin_param_var.shape)
table_shape[0] = zero_dim table_shape[0] = zero_dim
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册