PR #1291 Expose `dot_product_attention_weights`

Expose `dot_product_attention_weights` #1291

copybara-service merged 3 commits into google:master from avital:attention-weights

Expose `dot_product_attention_weights`

37d10fe4

google-cla added cla: yes

avital requested a review from

levskaya 4 years ago

avital removed review request from

levskaya 4 years ago

avital requested a review from

marcvanzee 4 years ago

avital requested a review from

levskaya 4 years ago

marcvanzee approved these changes on 2021-05-04

marcvanzee assigned

marcvanzee 4 years ago

marcvanzee added Priority: P1 - soon

Correct assertion string

43a98165

Add more assertions to dot_product_attention_weights

710d2023

avital added pull ready

copybara-service merged e95acab1 into master 4 years ago

Reviewers

marcvanzee

levskaya

Assignees

marcvanzee

Labels

cla: yes Priority: P1 - soon pull ready

Milestone

No milestone