tf.compat.v1.tpu.batch_parallel

Shards