tensorflow / 2.9.1 / distribute / inputreplicationmode.html /

tf.distribute.InputReplicationMode

Replication mode for input function.

  • PER_WORKER: The input function will be called on each worker independently, creating as many input pipelines as number of workers. Replicas will dequeue from the local Dataset on their worker. tf.distribute.Strategy doesn't manage any state sharing between such separate input pipelines.
  • PER_REPLICA: The input function will be called on each replica separately. tf.distribute.Strategy doesn't manage any state sharing between such separate input pipelines.
Class Variables
PER_REPLICA <InputReplicationMode.PER_REPLICA: 'PER_REPLICA'>
PER_WORKER <InputReplicationMode.PER_WORKER: 'PER_WORKER'>

© 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/InputReplicationMode