pytorch
e63db3ae - ENH Adds byte support for nll_loss (CUDA) (#60650)

Commit
4 years ago
ENH Adds byte support for nll_loss (CUDA) (#60650) Summary: Fixes https://github.com/pytorch/pytorch/issues/59765 This PR adds byte support for nll_loss on CUDA. Pull Request resolved: https://github.com/pytorch/pytorch/pull/60650 Reviewed By: albanD Differential Revision: D29429456 Pulled By: jbschlosser fbshipit-source-id: 894c969ed6bfc6117dee8e844a7cb5b99977247c
Author
Parents
Loading