pytorch

7ebe8328 - Address review comments on #28011.

Address review comments on #28011. Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/28109 Differential Revision: D17966067 fbshipit-source-id: 9e4a03a1813835b67cd614d8fac18524f5b36cc5