flax
a1a669aa - Improve the error message in MultiHeadDotProductAttention.

Commit

2 years ago

Improve the error message in MultiHeadDotProductAttention. PiperOrigin-RevId: 536390350

Author

a-googler

Committer

a-googler

Parents