tf.distribute.experimental.partitioners.MaxSizePartitioner