tree 2d0572f56d7821a2d1207fd9a1ea5d4074a0bcdc
parent 191fc8918cc29db9548f5a2c5c4ec1afae75e372
author Anish Tondwalkar <atondwal@google.com> 1652217231 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1652217390 -0700

[XLA:GPU] Add support for folding transpose of batch dimension into GeMMs.

Modifies `MatrixDescriptor`s to use two strides (for leading dim and batch), allowing matrices with interleaved batches. Relaxes the dot layout on GPU and transpose folding rules to support anything that can be expressed using those two strides.

PiperOrigin-RevId: 447825167