pytorch

27c22b9b - Modify function to takes dtype as argument

Modify function to takes dtype as argument Summary: To avoid repeating to() casts for every argument of the function Test Plan: CI Reviewed By: malfet Differential Revision: D22833521 fbshipit-source-id: ae0a8f70339cd6adfeea2f552d35bbcd48b11cf7