mlir-hlo/lib/Dialect/mhlo/transforms
Smit Hinsu 577a81a66d Sink standard dialect constants in sink_constants_to_control_flow pass
This is required before exporting HLO dialect ops with standard dialect constant to XLA.

Also, sink constants for sort op as well. Added a TODO to generalize this pass to handle more ops and non-const values defined outside.

PiperOrigin-RevId: 324301911
2020-08-03 19:30:29 -07:00
..
chlo_legalize_to_hlo.cc Integrate LLVM at llvm/llvm-project@cd4e8d7f6f 2020-08-03 19:29:29 -07:00
chlo_legalize_to_hlo_pass.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
hlo_legalize_to_lhlo.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
legalize_control_flow.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
legalize_gather_to_torch_index_select.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
legalize_tanh_to_approximation.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
legalize_to_linalg.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
legalize_to_standard.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
legalize_to_standard_patterns.td More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
lhlo_copy_removal.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
lhlo_fuse_linalg.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
lhlo_legalize_to_affine.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
lhlo_legalize_to_gpu.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
lhlo_legalize_to_llvm.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
lhlo_legalize_to_llvm_pass.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
lhlo_legalize_to_parallel_loops.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
lower_complex.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
lower_complex_patterns.td More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
lower_general_dot.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
materialize_broadcasts.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
materialize_broadcasts_pass.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
mhlo_fusion.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
optimize_mhlo.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
optimize_mhlo_pass.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
register_all_passes.cc Add optional static registration for mhlo/lmhlo passes. 2020-08-03 19:29:40 -07:00
sink_constants_to_control_flow.cc Sink standard dialect constants in sink_constants_to_control_flow pass 2020-08-03 19:30:29 -07:00
test_infer_shaped_type_pass.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
transform_unranked_hlo.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
unfuse_batch_norm.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
unfuse_batch_norm_pass.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00