[mlir][Tensor] Use output_shape for DimOp->ExpandShapeOp folding#118203

Groverkss wants to merge 1 commit intollvm:mainllvm/llvm-project:mainfrom

Groverkss:tensor-dim-folderGroverkss/llvm-project:tensor-dim-folderCopy head branch name to clipboard