Commit 36d3dc73 authored by Chris Jones's avatar Chris Jones Committed by TensorFlower Gardener
Browse files

Fix `TPUStrategy.extended.reduce_to` and allow user to specify destinations... 

Fix `TPUStrategy.extended.reduce_to` and allow user to specify destinations other than the TPU host.

PiperOrigin-RevId: 232685087
parent ac1919f6
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