Where is torch._C._nn.nll_loss?

1

I find the fold torch/csrc/nn is empty.

1 Like
2

The aten nn.yaml generates a binding for nll_loss, that eventually leads to the THNN ClassNLLCriterion or its cuda counterpart here.

5 Likes