Support range_iterator as a function input (#138657)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/138654
X-link: https://github.com/pytorch/pytorch/pull/138657
Approved by: https://github.com/williamwen42, https://github.com/jansel
Reviewed By: wdvr
Differential Revision: D64881833
Pulled By: anijain2305
fbshipit-source-id: 46bcffa12ef2bec0ff47a1b60323aacbb3a90872
