Support overridden __call__ on nn modules (#138619)
Summary:
X-link: https://github.com/pytorch/pytorch/pull/138619
Approved by: https://github.com/williamwen42
ghstack dependencies: #138657
Reviewed By: wdvr
Differential Revision: D64881836
Pulled By: anijain2305
fbshipit-source-id: 1974dbc228618e8597eb6ab293272ee985964f52
