[mlir][MemRef] Change memref-expand to be a generic operation pass

This pass doesn't rely on any specific characteristics of FuncOp, and
can just be a generic operation pass.

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