ENH Adds Byte support for nll_loss (CPU) (#60308)
Summary:
Addresses a part of https://github.com/pytorch/pytorch/issues/59765
This PR adds byte support for nll_loss on the CPU for `input.dim() == 2`.
CUDA support will be implemented when `nll_loss` migration to CUDA is completed in https://github.com/pytorch/pytorch/pull/60299 and https://github.com/pytorch/pytorch/pull/60097
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60308
Reviewed By: VitalyFedyunin
Differential Revision: D29329458
Pulled By: jbschlosser
fbshipit-source-id: d3585c4966030bc61e451f8aa817406a8a3acf47
