|
|
|
@ -552,10 +552,10 @@ class DistributeTranspiler(object):
|
|
|
|
op_on_pserver = True
|
|
|
|
op_on_pserver = True
|
|
|
|
new_outputs[key] = pserver_vars[op.output(key)[0]]
|
|
|
|
new_outputs[key] = pserver_vars[op.output(key)[0]]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if op_on_pserver:
|
|
|
|
# most startup program ops have no inputs
|
|
|
|
# most startup program ops have no inputs
|
|
|
|
new_inputs = self._get_input_map_from_op(pserver_vars, op)
|
|
|
|
new_inputs = self._get_input_map_from_op(pserver_vars, op)
|
|
|
|
|
|
|
|
|
|
|
|
if op_on_pserver:
|
|
|
|
|
|
|
|
if op.type in [
|
|
|
|
if op.type in [
|
|
|
|
"gaussian_random", "fill_constant", "uniform_random"
|
|
|
|
"gaussian_random", "fill_constant", "uniform_random"
|
|
|
|
]:
|
|
|
|
]:
|
|
|
|
|