tf.compat.v1.fixed_size_partitioner