flax
Align layernorm dtype handling with batchnorm (i.e., use requested dtype for layernorm outputs, even though intermediate computations are f32).
#331

Merged

Align layernorm dtype handling with batchnorm (i.e., use requested dtype for layernorm outputs, even though intermediate computations are f32). #331

copybara-service merged 1 commit into master from test_317419675

googlebot added cla: yes

copybara-service force pushed 5 years ago

Align layernorm dtype handling with batchnorm (i.e., use requested dt…

ed42d067

copybara-service force pushed to ed42d067 5 years ago

copybara-service merged ed42d067 into master 5 years ago

copybara-service deleted the test_317419675 branch 5 years ago

Reviewers

No reviews

Assignees

No one assigned

Labels

cla: yes

Milestone

No milestone