pytorch

bcbde490 - Fix flake (#34974)

Fix flake (#34974) Summary: fix flake, add overload names Pull Request resolved: https://github.com/pytorch/pytorch/pull/34974 Differential Revision: D20519191 Pulled By: eellison fbshipit-source-id: d08d36b64397287cad484690074e694d8a0e472e