Commit 8cb5ae9d authored by Sourabh Bajaj's avatar Sourabh Bajaj Committed by TensorFlower Gardener
Browse files

Add functionality to normalization layer to work in the cross-replica context 

PiperOrigin-RevId: 226068674
parent 2365c5c4
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment