ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


A container that holds a list of values, one value per worker.

tf.distribute.experimental.coordinator.PerWorkerValues contains a collection of values, where each of the values is located on its corresponding worker, and upon being used as one of the args or kwargs of tf.distribute.experimental.coordinator.ClusterCoordinator.schedule(), the value specific to a worker will be passed into the function being executed at that corresponding worker.

Currently, the only supported path to create an object of tf.distribute.experimental.coordinator.PerWorkerValues is through calling iter on a ClusterCoordinator.create_per_worker_dataset-returned distributed dataset instance. The mechanism to create a custom tf.distribute.experimental.coordinator.PerWorkerValues is not yet supported.