Commit 8cd3cbb1 authored by Dimitris Vardoulakis's avatar Dimitris Vardoulakis Committed by TensorFlower Gardener
Browse files

[XLA:TPU] Fix the algebra in ArCrsCombiner to take into account the number of... 

[XLA:TPU] Fix the algebra in ArCrsCombiner to take into account the number of replicas, not just the number of spatial partitions.

PiperOrigin-RevId: 224097074
parent be68e120
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