Combine logic for all pipeline optimizations that involve swapping stages into one function

Currently the pipeline optimizations that involve swapping stages are done in separate functions but with very similar logic, which could be abstracted out.