Solved by richard in post #2 
              
              
                The aten nn.yaml generates a binding for nll_loss, that eventually leads to the THNN ClassNLLCriterion or its cuda counterpart here.