Searched refs:_wrapped_loss_fn (Results 1 – 1 of 1) sorted by relevance
1074 loss_fn=self._wrapped_loss_fn,1113 def _wrapped_loss_fn(self, labels, logits, weights=None): member in _MultiClassHead