flax
a3a6f7ba - Concat the distributed batchnorm to make it more efficient.

Commit

5 years ago

Concat the distributed batchnorm to make it more efficient. PiperOrigin-RevId: 315580269

References

Author

a-googler

Committer

a-googler

Parents