Parameterized partition_size