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: Synchronous training on TPUs and TPU Pods.
class ValueContext: A class wrapping information needed by a distribute function.