flax
Fix promotion bug in MultiHeadDotProductAttention:
#3571

Merged

Fix promotion bug in MultiHeadDotProductAttention: #3571

copybara-service merged 1 commit into google:main from giovannic:bug/attention_promotion

giovannic marked this pull request as ready for review 2 years ago

cgarciae commented on 2023-12-18

Fix promotion bug in MultiHeadDotProductAttention:

1c27546e

giovannic force pushed to 1c27546e 2 years ago

cgarciae approved these changes on 2023-12-18

cgarciae added pull ready

copybara-service merged 85dfad24 into main 2 years ago

giovannic deleted the bug/attention_promotion branch 2 years ago

Reviewers

cgarciae

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone