tf.distribute.experimental.coordinator.PerWorkerValues
A container that holds a list of values, one value per worker.
tf.distribute.experimental.coordinator.PerWorkerValues(
values
)
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.
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/distribute/experimental/coordinator/PerWorkerValues
