Searched refs:nll_loss_helper (Results 1 – 1 of 1) sorted by relevance
2834 def nll_loss_helper(input, target, weight, ignore_index): function2841 losses_and_weights = [nll_loss_helper(i, t, weight, ignore_index)