tf.distribute.experimental.partitioners.Partitioner