Fix AT_USE_JITERATOR checks (#100464)
`AT_USE_JITERATOR` evaluates to false when the definition isn't
included, so these files were not using jiterator at all.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/100464
Approved by: https://github.com/lezcano, https://github.com/ngimel
