flax
Adds axis_name and axis_index_groups to LayerNorm and GroupNorm.
#2402
Merged

Adds axis_name and axis_index_groups to LayerNorm and GroupNorm. #2402

copybara-service merged 0 commits into main from test_467990868
copybara-service
google-cla
copybara-service copybara-service changed the title Adds axis_name and axis_index_groups to LayerNorm. Adds axis_name and axis_index_groups to LayerNorm and GroupNorm. 3 years ago
copybara-service copybara-service force pushed from 459ed781 3 years ago
codecov-commenter
copybara-service copybara-service merged 2b73efdd into main 3 years ago
copybara-service copybara-service force pushed to 2b73efdd 3 years ago
copybara-service copybara-service deleted the test_467990868 branch 3 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone