pytorch

36d4062a - Fix some variable types (#60414)

Fix some variable types (#60414) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60414 Test Plan: Sandcastle Reviewed By: ngimel Differential Revision: D29221183 fbshipit-source-id: f855efca2fd08844de65d0f9ef73bcceffee657e