flax
Make `force_fp32_for_softmax` arg in `MultiHeadDotProductAttention` useful.
#4029

Merged

Make `force_fp32_for_softmax` arg in `MultiHeadDotProductAttention` useful. #4029

copybara-service merged 1 commit into main from test_646659942

copybara-service force pushed 1 year ago

Make `force_fp32_for_softmax` arg in `MultiHeadDotProductAttention` u…

9e6b8193

copybara-service force pushed to 9e6b8193 1 year ago

copybara-service merged 9e6b8193 into main 1 year ago

copybara-service deleted the test_646659942 branch 1 year ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone