Use c10::irange for great good (#52153)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52153
Test Plan: Sandcastle
Reviewed By: ngimel
Differential Revision: D26407087
fbshipit-source-id: ea8ce1c17299cb9d89621e4a39f31edc2faa9fd6
