aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/service/batch_dot_simplification_test.cc
Commit message (Collapse)AuthorAge
* Convert a couple more test files to HloVerifiedTestBase, and add default ↵Gravatar Dimitris Vardoulakis2018-08-29
| | | | | | arguments to the constructor to remove some boilerplate. PiperOrigin-RevId: 210855509
* [XLA] Add and use a layout-sensitive HLO verifier.Gravatar Justin Lebar2018-08-24
| | | | | | | | | | | | | | For now, this verifier checks some noncontroversial invariants, like: - Fusion operands and fusion computation parameters must have matching layouts. - Same for while loops, calls, kConditional. It's a bit of a pain to add these explicit layout-sensitive and allow-mixed-precision flags everywhere, but I think it's better than adding default args. With default args we can easily mix up the order, and we'd only be able to add new flags to the end of the list. PiperOrigin-RevId: 210059349
* Remove degenerate batch dimensions form batch dotGravatar Sanjoy Das2018-05-11
The way things are set up today this specific optimization isn't particularly important, but I want to implement a follow-on optimization in BatchDotSimplification to transform (non-degenerate) batch GEMV operations into GEMM which I'm expecting to help us a bit. This would normally be in the algebraic simplifier, but we want to fixpoint this pass before we run DotDecomposer. This will become more important when we implement the (non-degenerate) batch GEMV operations -> GEMM transform. PiperOrigin-RevId: 196314230