* [MLIR] Modify mlir subgraph extraction pass.

* Address PR feedback.

* Modify sub-graph construction algorithm.

* Address PR feedback.

* Address PR feedback.

* Change macro to function.