Searched refs:_label_ids (Results 1 – 1 of 1) sorted by relevance
1109 def _label_ids(self, labels_dict): member in _MultiClassHead1114 return self._loss_fn(self._label_ids(labels), logits, weights=weights)1146 values=((eval_loss, self._labels(labels), self._label_ids(labels),1169 self._label_ids(labels), weights, class_id))