mlir-hlo/lib/Dialect/mhlo/transforms
Stephan Herhut 2aa07b0091 Insert explicit casts to model extra shape knowledge for unranked chlo transform
When transforming unranked binary operations from CHLO to HLO, we insert `shape.broadcast` operations. Due to context, we know that the result of the `shape.broadcast` operation has a static shape. Instead of modelling this in the type of the broadcast operation itself, which is illegal,  we now use an explicit cast.

PiperOrigin-RevId: 331989879
2020-09-16 06:14:49 -07:00
..
CMakeLists.txt [XLA][MLIR] Lower `tf.Tan` and `tf.Sin` to MLHLO 2020-09-14 02:34:52 -07:00
chlo_legalize_to_hlo.cc Insert explicit casts to model extra shape knowledge for unranked chlo transform 2020-09-16 06:14:49 -07:00
chlo_legalize_to_hlo_pass.cc Remove the dependency on global dialect registry from mlir-hlo 2020-08-25 20:30:42 -07:00
chlo_legalize_to_hlo_patterns.td [XLA][MLIR] Lower `tf.Tan` and `tf.Sin` to MLHLO 2020-09-14 02:34:52 -07:00
hlo_legalize_to_lhlo.cc [XLA][MLIR] Lower `tf.Tan` and `tf.Sin` to MLHLO 2020-09-14 02:34:52 -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 Add support for legalizing lmhlo.transpose to linalg.generic 2020-09-04 15:37:42 -07:00
legalize_to_standard.cc Integrate LLVM at llvm/llvm-project@202766947e 2020-09-02 02:27:52 -07:00
legalize_to_standard_patterns.td Lowering for mhlo.ceil to std.ceil 2020-08-12 16:15:35 -07:00
lhlo_fuse_linalg.cc Remove the dependency on global dialect registry from mlir-hlo 2020-08-25 20:30:42 -07:00
lhlo_legalize_to_affine.cc Remove the dependency on global dialect registry from mlir-hlo 2020-08-25 20:30:42 -07:00
lhlo_legalize_to_gpu.cc Remove the dependency on global dialect registry from mlir-hlo 2020-08-25 20:30:42 -07:00
lhlo_legalize_to_llvm.cc Integrate LLVM at llvm/llvm-project@52f0837778 2020-09-10 08:14:23 -07:00
lhlo_legalize_to_llvm_pass.cc Remove the dependency on global dialect registry from mlir-hlo 2020-08-25 20:30:42 -07:00
lhlo_legalize_to_parallel_loops.cc Remove the dependency on global dialect registry from mlir-hlo 2020-08-25 20:30:42 -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_control_flow_to_scf.cc Integrate LLVM at llvm/llvm-project@202766947e 2020-09-02 02:27:52 -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
sink_constants_to_control_flow.cc Internal change 2020-08-03 22:59:44 -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 [MLIR] Simplify and generalize `transform-unranked-hlo` 2020-09-16 01:13:23 -07:00
unfuse_batch_norm.cc Integrate LLVM at llvm/llvm-project@202766947e 2020-09-02 02:27:52 -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