mlir-hlo/lib/Dialect/mhlo/transforms
Robert Suderman 98a1e3b108 Add an optimization that converts some Gathers to Slices.
Some Gathers can be represented as slices. This lowering transforms
these gathers into slices.

PiperOrigin-RevId: 321394868
2020-07-30 22:34:10 +00:00
..
canonicalize.td Move the HLO/LHLO dialects to a new directory: tensorflow/compiler/mlir/hlo 2020-07-30 22:32:32 +00:00
chlo_legalize_to_hlo.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
chlo_legalize_to_hlo_pass.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
hlo_legalize_to_lhlo.cc Rename XlaHloDialect class into MhloDialect following the recent dialect namespace renaming 2020-07-30 22:33:20 +00:00
legalize_control_flow.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
legalize_tanh_to_approximation.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
legalize_to_linalg.cc PR #40925: [MLIR] Update lhlo.const to linalg lowering to use affine.store inste… 2020-07-30 22:33:51 +00:00
legalize_to_standard.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
legalize_to_standard_patterns.td Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
lhlo_copy_removal.cc Rename xla_lhlo dialect into lmhlo 2020-07-30 22:33:11 +00:00
lhlo_fuse_linalg.cc Rename xla_lhlo dialect into lmhlo 2020-07-30 22:33:11 +00:00
lhlo_legalize_to_affine.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
lhlo_legalize_to_gpu.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
lhlo_legalize_to_llvm.cc Implement lowering of lmhlo.reshape_memref_cast to LLVM for unknown length shape operand. 2020-07-30 22:34:02 +00:00
lhlo_legalize_to_llvm_pass.cc Add mlir_cpu_runner tests infra for CHLO->LHLO->LLVM lowering. 2020-07-30 22:33:25 +00:00
lhlo_legalize_to_parallel_loops.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
lower_complex.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
lower_complex_patterns.td Move XLA-independent transforms to the new MLIR-HLO directory 2020-07-30 22:32:40 +00:00
lower_general_dot.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
materialize_broadcasts.cc Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
materialize_broadcasts_pass.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
mhlo_fusion.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
optimize_mhlo.cc Add an optimization that converts some Gathers to Slices. 2020-07-30 22:34:10 +00:00
optimize_mhlo_pass.cc Add an optimization that converts some Gathers to Slices. 2020-07-30 22:34:10 +00:00
sink_constants_to_control_flow.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
test_infer_shaped_type_pass.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
transform_unranked_hlo.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
unfuse_batch_norm.cc Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
unfuse_batch_norm_pass.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00