Searched refs:logits_to_predictions_fn (Results 1 – 1 of 1) sorted by relevance
633 logits_to_predictions_fn, argument651 predictions = logits_to_predictions_fn(logits)749 logits_to_predictions_fn=self._logits_to_predictions,870 logits_to_predictions_fn=self._logits_to_predictions,1075 logits_to_predictions_fn=self._logits_to_predictions,1272 logits_to_predictions_fn=self._logits_to_predictions,1372 logits_to_predictions_fn=self._logits_to_predictions,