Searched refs:_masked_softmax (Results 1 – 1 of 1) sorted by relevance
417 def _masked_softmax(self, attention_scores, attention_mask=None): member in MultiHeadAttention464 attention_scores = self._masked_softmax(attention_scores, attention_mask)