Please see the definition of these values in TPUConfig.
BROADCAST = 4
PER_HOST_V1 = 2
PER_HOST_V2 = 3
PER_SHARD_V1 = 1
SLICED = 5
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/compat/v1/estimator/tpu/InputPipelineConfig