pytorch

ad104157 - Fix loop types (#60504)

Fix loop types (#60504) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60504 Test Plan: Sandcastle Reviewed By: ngimel Differential Revision: D29313197 fbshipit-source-id: bc86622b587e4fdb73431c2ff27300404c9693ae