Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tf.compat.v2.distribute.experimental

TensorFlow 2.0 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 MultiWorkerMirroredStrategy: Distribution strategy that uses collective ops for all-reduce.

class ParameterServerStrategy: An asynchronous multi-worker parameter server DistributionStrategy.

class TPUStrategy: TPU distribution strategy implementation.