Commit 3f822367 authored by Peter Hawkins's avatar Peter Hawkins Committed by TensorFlower Gardener
Browse files

[TPU] Register gradient for CrossReplicaSum. Happily, the gradient for... 

[TPU] Register gradient for CrossReplicaSum. Happily, the gradient for CrossReplicaSum is just a CrossReplicaSum.

PiperOrigin-RevId: 172132628
parent cbb4d188
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