flax
2b73efdd
- Adds axis_name and axis_index_groups to LayerNorm and GroupNorm.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
3 years ago
Adds axis_name and axis_index_groups to LayerNorm and GroupNorm. PiperOrigin-RevId: 468499562
References
#2402 - Adds axis_name and axis_index_groups to LayerNorm and GroupNorm.
Author
a-googler
Committer
a-googler
Parents
c3e5363d
Loading
