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

Find file Blame History Permalink

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

attention_test.py 10.7 KB