diff options
| author |  Yuefeng Zhou <yuefengz@google.com> | 2018-08-03 18:07:17 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-03 18:11:18 -0700 |
| commit | 76c9af7e37709015ed51ee828010dcee925eb12e (patch) | |
| tree | 5f56d322788ac93c62c281790e0c8fce73f9bfab /tensorflow/contrib/distribute/__init__.py | |
| parent | ea5b1e4b5e3f1f8682c18f6ad09af45eb17f415d (diff) | |
Add multi-worker strategies to __init__.py.
PiperOrigin-RevId: 207354363
Diffstat (limited to 'tensorflow/contrib/distribute/__init__.py')
| -rw-r--r-- | tensorflow/contrib/distribute/__init__.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/contrib/distribute/__init__.py b/tensorflow/contrib/distribute/__init__.py index 2e2c3be853..9123ca749b 100644 --- a/tensorflow/contrib/distribute/__init__.py +++ b/tensorflow/contrib/distribute/__init__.py @@ -19,10 +19,13 @@ from __future__ import division from __future__ import print_function # pylint: disable=unused-import,wildcard-import +from tensorflow.contrib.distribute.python.collective_all_reduce_strategy import CollectiveAllReduceStrategy from tensorflow.contrib.distribute.python.cross_tower_ops import * from tensorflow.contrib.distribute.python.mirrored_strategy import MirroredStrategy +from tensorflow.contrib.distribute.python.multi_worker_strategy import MultiWorkerMirroredStrategy from tensorflow.contrib.distribute.python.monitor import Monitor from tensorflow.contrib.distribute.python.one_device_strategy import OneDeviceStrategy +from tensorflow.contrib.distribute.python.parameter_server_strategy import ParameterServerStrategy from tensorflow.contrib.distribute.python.step_fn import * from tensorflow.contrib.distribute.python.tpu_strategy import TPUStrategy from tensorflow.python.training.distribute import * @@ -32,11 +35,14 @@ from tensorflow.python.util.all_util import remove_undocumented _allowed_symbols = [ 'AllReduceCrossTowerOps', + 'CollectiveAllReduceStrategy', 'CrossTowerOps', 'DistributionStrategy', 'MirroredStrategy', + 'MultiWorkerMirroredStrategy', 'Monitor', 'OneDeviceStrategy', + 'ParameterServerStrategy', 'ReductionToOneDeviceCrossTowerOps', 'Step', 'StandardInputStep', |