flax
b964fb9a - Add use_fast_variance option to GroupNorm and BatchNorm to allow disabling it.

Commit

2 years ago

Add use_fast_variance option to GroupNorm and BatchNorm to allow disabling it. PiperOrigin-RevId: 553781414

Author

a-googler

Committer

a-googler

Parents