[mlir][scf] Fold dim(scf.for) to dim(iter_arg)

Fold dim ops of scf.for results to dim ops of the respective iter args if the loop is shape preserving.

Differential Revision: https://reviews.llvm.org/D109430