flax
changed `return_weights` to `sow_weights` for attention layer
#3550

Merged

changed `return_weights` to `sow_weights` for attention layer #3550

copybara-service merged 1 commit into google:main from chiamp:attention

changed to for attention layer

7f10ad8d

chiamp assigned

chiamp 2 years ago

chiamp requested a review from

cgarciae 2 years ago

cgarciae approved these changes on 2023-12-14

cgarciae added pull ready

copybara-service merged 2414d1cc into main 2 years ago

chiamp deleted the attention branch 2 years ago

Reviewers

cgarciae

Assignees

chiamp

Labels

pull ready

Milestone

No milestone