[llvm] [mlir] [MLIR] Generalize expand_shape to take shape as explicit input (PR #69267)

[via llvm-commits]

https://github.com/MaheshRavishankar requested changes to this pull request.

Thanks! Left a few mostly minor comments. Nothing that I see as blocker for landing.

https://github.com/llvm/llvm-project/pull/69267