pytorch

ed63fb52 - Fix some more loops (#60895)

Fix some more loops (#60895) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60895 Test Plan: Sandcastle Reviewed By: ngimel Differential Revision: D29431572 fbshipit-source-id: fbcf48696bf2c90cc0973a767d83bb526f6ccd7f