Add a downward-only shuffle partition pass sized from consumer-stage shuffle input

Add a downward-only shuffle partition pass sized from consumer-stage shuffle input #2146