mlir-hlo/lib/Dialect/mhlo/transforms
Adrian Kuegel 4c282fb542 Avoid Broadcast op if all shapes are (known to) be equal.
The rank specialization case for shapes which are either of the same shape or a
scalar doesn't need to compute the final result shape.

PiperOrigin-RevId: 380129316
2021-06-18 00:22:35 -07:00
..
CMakeLists.txt [MLIR][HLO] Remove unused pass `TransformUnrankedHloPass` 2021-06-17 05:20:49 -07:00
broadcast_propagation.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00
chlo_legalize_to_hlo.cc [MLIR][HLO] Remove unused pass `TransformUnrankedHloPass` 2021-06-17 05:20:49 -07:00
chlo_legalize_to_hlo_pass.cc Separate CHLO transforms for expanding compositions and lowering broadcasts. 2021-05-18 13:33:59 -07:00
chlo_legalize_to_hlo_patterns.td Fix Cosh approximation for F16. 2021-06-10 06:16:44 -07:00
fusion_utils.cc PR #50020: [MLIR][DISC] support fusion on buffer 2021-06-16 09:51:29 -07:00
hlo_legalize_to_lhlo.cc PR #50271: [MLIR][DISC] Bufferize GatherOp and DynamicGatherOp 2021-06-16 13:47:56 -07:00
legalize_control_flow.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00
legalize_gather_to_torch_index_select.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00
legalize_tensor_load_op.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00
legalize_to_linalg.cc Integrate LLVM at llvm/llvm-project@366df11a35 2021-06-17 17:29:07 -07:00
legalize_to_standard.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00
legalize_to_standard_patterns.td Lowering for mhlo.ceil to std.ceil 2020-08-12 16:15:35 -07:00
legalize_trigonometric_to_approximation.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00
lhlo_fuse_linalg.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00
lhlo_fusion.cc PR #50020: [MLIR][DISC] support fusion on buffer 2021-06-16 09:51:29 -07:00
lhlo_legalize_to_affine.cc PR #50073: [MLIR] Add GatherOp lowering from lmhlo to Affine. 2021-06-17 14:55:49 -07:00
lhlo_legalize_to_gpu.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00
lhlo_legalize_to_parallel_loops.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00
lower_complex.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00
lower_complex_patterns.td Generate Equal and NotEqual kernels for complex types. 2021-04-15 00:35:52 -07:00
lower_general_dot.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -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 Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00
mhlo_control_flow_to_scf.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00
mhlo_fusion.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -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 Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00
ral_inject_execution_context.cc PR #50191: [MLIR][DISC] Add RAL (Runtime abstraction layer) Dialect 2021-06-14 11:27:43 -07:00
rank_specialization.cc Avoid Broadcast op if all shapes are (known to) be equal. 2021-06-18 00:22:35 -07:00
sink_constants_to_control_flow.cc [MLIR][HLO] Annotate `mhlo.clamp` and `mhlo.select` as element-wise broadcasting 2021-06-16 07:59:26 -07:00
test_infer_shaped_type_pass.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00
unfuse_batch_norm.cc Integrate LLVM at llvm/llvm-project@678241795c 2021-03-16 13:33:00 -07:00
unfuse_batch_norm_pass.cc Fix pass definition to inherit from the TableGen generated base class (NFC) 2021-06-16 19:05:11 -07:00