Google I/O returns May 18-20! Reserve space and build your schedule Register now

Module: tf.compat.v1.distribute.experimental

Public API for tf.distribute.experimental namespace.


class CentralStorageStrategy: A one-machine strategy that puts all variables on a single device.

class CollectiveCommunication: Cross device communication implementation.

class CollectiveHints: Hints for collective operations like AllReduce.

class CommunicationImplementation: Cross device communication implementation.

class CommunicationOptions: Options for cross device communications like All-reduce.

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.