flax
Remove Implicit rank promotion
#1535
Merged

Remove Implicit rank promotion #1535

jheek
jheek Remove Implicit rank promotion
a401e824
jheek jheek requested a review from avital avital 4 years ago
jheek jheek requested a review from levskaya levskaya 4 years ago
google-cla google-cla added cla: yes
jheek Fix seq2seq implicit rank promotion
ebe84238
jheek Fix implicit rank promotion in sst2
bc3ae72a
jheek Fix layernorm rank promotion
6f749e00
jheek Fix one hot implicit rank promotion
05fe7096
jheek Fix temp sample implicit rank promo
1fc7bfe9
levskaya
levskaya approved these changes on 2021-09-11
levskaya levskaya added pull ready
copybara-service copybara-service merged 19c4e273 into main 4 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone