From 7330cd639cf72acd3b5fa80ec880fe3f82dc4467 Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Fri, 26 Apr 2019 13:06:44 +0800 Subject: [PATCH] truncated_gaussian_random supported in distributed training, test=develop (#17091) --- python/paddle/fluid/transpiler/distribute_transpiler.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/transpiler/distribute_transpiler.py b/python/paddle/fluid/transpiler/distribute_transpiler.py index e6963f5461e..60f74bb6264 100644 --- 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( -- GitLab