[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
Loading
Please sign in to comment