pytorch

ef0baba2 - Use `int64_t` for nll_loss with cuda inputs (#85395)

Use `int64_t` for nll_loss with cuda inputs (#85395) Related #85005 Pull Request resolved: https://github.com/pytorch/pytorch/pull/85395 Approved by: https://github.com/t-vi, https://github.com/lezcano