Support non-PerDevice value to be reduced in cross_tower_ops.
This allows MirroredStrategy with one device and between-graph strategies with one tower on each worker. PiperOrigin-RevId: 209884569
Loading
Please sign in to comment
This allows MirroredStrategy with one device and between-graph strategies with one tower on each worker. PiperOrigin-RevId: 209884569