patching graph_for (#55139)
Summary:
Allows individual DifferentiableGraphOp to display optimized forward graph. This improves user visibility to graph mutation via optimization pass, especially fusion.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/55139
Reviewed By: albanD
Differential Revision: D31330909
Pulled By: dzhulgakov
fbshipit-source-id: c745b482fdc34876dc404cbe3bacd99dcf2ac724