tree 9110bcd311bf6c02cda0ba941f222baf65121e99
parent b2a844933cd21f9daeacd9f3210d2cbe4851ee8d
author A. Unique TensorFlower <gardener@tensorflow.org> 1621632021 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1621632805 -0700

[XLA:GPU] Enable SPMD passes even when num_partitions == 1.
- This won't do any partitioning as such, but will eliminate the "Sharding" calls that were
  added in the IR.

PiperOrigin-RevId: 375162846
Change-Id: Idae3f2f0a71d26c8adc6eac8381176b1af0c9516