tf.distribute.experimental.partitioners.MinSizePartitioner