Searched refs:_softmax_cross_entropy_loss (Results 1 – 3 of 3) sorted by relevance
70 return head_lib._softmax_cross_entropy_loss( # pylint: disable=protected-access
92 loss_fn = _softmax_cross_entropy_loss418 def _softmax_cross_entropy_loss(logits, target): function
964 def _softmax_cross_entropy_loss(labels, logits, weights=None): function1044 self._loss_fn = loss_fn if loss_fn else _softmax_cross_entropy_loss