Searched refs:_mean_squared_loss (Results 1 – 3 of 3) sorted by relevance
52 return head_lib._mean_squared_loss( # pylint: disable=protected-access
55 loss_fn=_mean_squared_loss,399 def _mean_squared_loss(logits, target): function
226 loss_fn=_mean_squared_loss,559 def _mean_squared_loss(labels, logits, weights=None): function