Skip to content

Fix use_causal_mask=True with RaggedTensor bug#17231

Merged
copybara-service[bot] merged 1 commit intokeras-team:masterfrom
haifeng-jin:mha
Nov 9, 2022
Merged

Fix use_causal_mask=True with RaggedTensor bug#17231
copybara-service[bot] merged 1 commit intokeras-team:masterfrom
haifeng-jin:mha

Conversation

@haifeng-jin
Copy link
Copy Markdown
Contributor

Before this PR, it crashes when passing RaggedTensor to MultiHeadAttention layer with use_causal_mask=True.

@haifeng-jin haifeng-jin added the ready to pull Ready to be merged into the codebase label Nov 9, 2022
@gbaned gbaned requested a review from fchollet November 9, 2022 14:09
@google-ml-butler google-ml-butler bot added the keras-team-review-pending Pending review by a Keras team member. label Nov 9, 2022
@copybara-service copybara-service bot merged commit 29f5006 into keras-team:master Nov 9, 2022
@haifeng-jin haifeng-jin deleted the mha branch November 9, 2022 20:45
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

keras-team-review-pending Pending review by a Keras team member. ready to pull Ready to be merged into the codebase size:S

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants