Remove deprecated ps_devices argument.

afrozenator · Copybara-Service · commit 6def2ceae17a · 2018-10-29T16:41:47.000-07:00
PiperOrigin-RevId: 219215973
diff --git a/tensor2tensor/models/research/transformer_moe.py b/tensor2tensor/models/research/transformer_moe.py
@@ -106,7 +106,6 @@ def decorated(x, *args, **kwargs):
     layers = common_attention.get_standardized_layers(
         hparams=hparams,
         dp=dp,
-        ps_devices=self._ps_devices,
     )
 
     if hparams.mode == tf.estimator.ModeKeys.TRAIN:

Original file line number	Diff line number	Diff line change
`@@ -106,7 +106,6 @@ def decorated(x, args, *kwargs):`
`106`	`106`	`layers = common_attention.get_standardized_layers(`
`107`	`107`	`hparams=hparams,`
`108`	`108`	`dp=dp,`
`109`		`- ps_devices=self._ps_devices,`
`110`	`109`	`)`
`111`	`110`
`112`	`111`	`if hparams.mode == tf.estimator.ModeKeys.TRAIN:`