Use stage input partitioning partitionCount instead of output partitioning#16052

raunaqmorarka merged 2 commits intotrinodb:mastertrinodb/trino:masterfrom

gaurav8297:fix_partition_countCopy head branch name to clipboard