This is a follow up issue to track the feedback on this PR (https://github.com/pytorch/executorch/pull/13296) and potential future refactor Simplify this by using a subgraph_rewriter? see - https://docs.pytorch.org/executorch/stable/compiler-custom-compiler-passes.html#level-2