flax
b4591c14 - [linen] More minor cleanup in normalization `compute_stats`.

Commit

2 years ago

[linen] More minor cleanup in normalization `compute_stats`. PiperOrigin-RevId: 549515492

References

Author

chr1sj0nes

Committer

a-googler

Parents