official/nlp/modeling/layers/attention_test.py · 055acc0f444e0ceea54811802fcb0bb5d5fe3b29 · ModelZoo / ResNet50_tensorflow · GitLab

Find file Blame History Permalink

Make sure the attention mask expand dim happens at the head dim. · 33fc3eeb
Hongkun Yu authored Aug 07, 2020
```
PiperOrigin-RevId: 325454896
```
33fc3eeb

attention_test.py 10.7 KB