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

6 years ago · 7330cd639c
parent 794a195881
commit 7330cd639c
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/fluid/transpiler/distribute_transpiler.py
+++ b/python/paddle/fluid/transpiler/distribute_transpiler.py
@ -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(