✨[Feature] nn.module-defined atomic subgraph

[cehongwang]

Is your feature request related to a problem? Please describe.

Describe the solution you'd like

Describe alternatives you've considered

Additional context