TensorFlow 1 version |
Experimental Distribution Strategy library.
Classes
class CentralStorageStrategy
: A one-machine strategy that puts all variables on a single device.
class CollectiveCommunication
: Communication choices for CollectiveOps.
class CollectiveHints
: Hints for collective operations like AllReduce.
class MultiWorkerMirroredStrategy
: A distribution strategy for synchronous training on multiple workers.
class ParameterServerStrategy
: An asynchronous multi-worker parameter server tf.distribute strategy.
class TPUStrategy
: TPU distribution strategy implementation.