Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/24384 So that we can use them in other functions. Reviewed By: yinghai Differential Revision: D16824289 fbshipit-source-id: 3cb33cfa9a5c479a63db6438aef518209bdfb1f4