flax
Unit tests for `attention_bias` parameter to `MultiHeadDotProductAttention`. Add parameter to all overloads to make pytype happy.
#4702
Merged

Unit tests for `attention_bias` parameter to `MultiHeadDotProductAttention`. Add parameter to all overloads to make pytype happy. #4702

copybara-service merged 1 commit into main from test_747028285
copybara-service
copybara-service copybara-service force pushed from e0053d2e to fbdb96f9 1 year ago
copybara-service copybara-service changed the title Unit tests for `attention_bias` parameter to `MultiHeadDotProductAttention`. Unit tests for `attention_bias` parameter to `MultiHeadDotProductAttention`. Add parameter to all overloads to make pytype happy. 1 year ago
copybara-service copybara-service force pushed from fbdb96f9 to ec9f0596 1 year ago
copybara-service copybara-service force pushed from ec9f0596 to cc88b2b9 1 year ago
Unit tests for `attention_bias` parameter to `MultiHeadDotProductAtte…
5bba7b45
copybara-service copybara-service force pushed from cc88b2b9 to 5bba7b45 1 year ago
copybara-service copybara-service merged 5bba7b45 into main 1 year ago
copybara-service copybara-service deleted the test_747028285 branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone