Searched refs:_split_logits (Results 1 – 1 of 1) sorted by relevance
1654 head_logits_pairs = zip(self._heads, self._split_logits(logits))1676 def _split_logits(self, logits): member in _MultiHead