tree 4019f0b953ccd444201e10c5ab7478d3870b98ac
parent 8e5ab11c65e08e7b04505a0c965d01b6330810c2
author Marcello Maggioni <maggioni@google.com> 1614762114 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1614762418 -0800

[XLA] Make the canonicalize all-gather pass add a new/exclusive all-gather dimension.

Make the canonicalizer always add a completely new all gather dimension to the shape
of the all-gather operand to help layout assignment.

PiperOrigin-RevId: 360612458
Change-Id: I8299301219f704f14661f6fb2619ef604bb4d7c4