Searched refs:LOGITS (Results 1 – 3 of 3) sorted by relevance
33 LOGITS = "logits" variable in PredictionKey
902 prediction_key.PredictionKey.LOGITS:1135 prediction_key.PredictionKey.LOGITS: logits,1148 logits = predictions[prediction_key.PredictionKey.LOGITS]1296 prediction_key.PredictionKey.LOGITS:1397 prediction_key.PredictionKey.LOGITS:1415 logits = predictions[prediction_key.PredictionKey.LOGITS]
1808 ("head1", prediction_key.PredictionKey.LOGITS),1811 ("head2", prediction_key.PredictionKey.LOGITS),