Make force_fp32_for_softmax arg in MultiHeadDotProductAttention useful. #4029
Merged
copybara-service[bot] merged 1 commit intomainfrom test_646659942Jun 26, 2024
+22-29
force_fp32_for_softmax arg in MultiHeadDotProductAttention useful. #4029
