Searched refs:_calculate_scores (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | dense_attention.py | 87 def _calculate_scores(self, query, key): member in BaseDenseAttention
157 scores = self._calculate_scores(query=q, key=k)
342 def _calculate_scores(self, query, key): member in Attention
485 def _calculate_scores(self, query, key): member in AdditiveAttention
|
| D | dense_attention_test.py | 185 actual = attention_layer._calculate_scores(query=q, key=k)
201 actual = attention_layer._calculate_scores(query=q, key=k)
221 actual = attention_layer._calculate_scores(query=q, key=k)
238 actual = attention_layer._calculate_scores(query=q, key=k)
536 actual = attention_layer._calculate_scores(query=q, key=k)
554 actual = attention_layer._calculate_scores(query=q, key=k)
578 actual = attention_layer._calculate_scores(query=q, key=k)
|