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