Searched refs:_compute_attention (Results 1 – 2 of 2) sorted by relevance
429 def _compute_attention(self, member in MultiHeadAttention499 attention_output, attention_scores = self._compute_attention(
250 def _compute_attention(self, member in SubclassAttention