Trace directly into first-class module form. (#19722)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19722
ghimport-source-id: b024666feccb324f5ba9aae4a6301723e04d9846
Reviewed By: jamesr66a
Differential Revision: D15078535
Pulled By: zdevito
fbshipit-source-id: b866b31c1864a090c545560cbecee81e34ad2d16