Skip to content

Add return_attention_scores support to CachedMultiHeadAttention - #2213

Closed
DakshBegani wants to merge 6 commits into
keras-team:masterfrom
DakshBegani:fix/return-attention-scores
Closed

Add return_attention_scores support to CachedMultiHeadAttention#2213
DakshBegani wants to merge 6 commits into
keras-team:masterfrom
DakshBegani:fix/return-attention-scores

Fix test to handle potential None return from attention_scores

70c4464
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs