未验证 提交 7330cd63 编写于 作者: T tangwei12 提交者: GitHub

truncated_gaussian_random supported in distributed training, test=develop (#17091)

上级 794a1958
......@@ -1017,7 +1017,8 @@ class DistributeTranspiler(object):
new_inputs = self._get_input_map_from_op(pserver_vars, op)
if op.type in [
"gaussian_random", "fill_constant", "uniform_random"
"gaussian_random", "fill_constant", "uniform_random",
"truncated_gaussian_random"
]:
op._set_attr("shape", list(new_outputs["Out"].shape))
s_prog.global_block().append_op(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册