pytorch

f1775796 - Fix minor issues with #21736 (#22074)

Fix minor issues with #21736 (#22074) Summary: cc mrshenli Pull Request resolved: https://github.com/pytorch/pytorch/pull/22074 Differential Revision: D15965376 Pulled By: mrshenli fbshipit-source-id: 50ff96de6390817d8ea52c04322c6bee3d649b32