Add `partition_all`