flax
Avoid using float32 in normalization for mean/var and scale/bias parameters when force_float32_reductions=False
#4314
Merged

Avoid using float32 in normalization for mean/var and scale/bias parameters when force_float32_reductions=False #4314

copybara-service merged 1 commit into main from test_687929450
copybara-service
copybara-service copybara-service force pushed 1 year ago
copybara-service copybara-service force pushed 1 year ago
copybara-service copybara-service force pushed 1 year ago
eunjaekim-0 Avoid using float32 in normalization for mean/var and scale/bias para…
40f080ed
copybara-service copybara-service force pushed to 40f080ed 1 year ago
copybara-service copybara-service merged 40f080ed into main 1 year ago
copybara-service copybara-service deleted the test_687929450 branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone