Searched refs:_build_attention (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ | ||
| D | multi_head_attention.py | 379 self._build_attention(output_rank) |
| D | multi_head_attention_test.py | 247 def _build_attention(self, qkv_rank): member in SubclassAttention |