mlir-hlo/lib/Dialect/mhlo/transforms
Rahul Joshi 1958f228ec [MLIR:HLO] Extend CustomCall to support multiple outputs.
- Extend MHLO CustomCall to have multiple tensors as results.
- Extend LHLO CustomCall to have multiple memrefs for output operands.
- Fix HLO->LHLO and XLA HLO->LHLO mapping for CustomCall to setup the
  operand_segment_sizes attribute correctly.

PiperOrigin-RevId: 342067762
2020-11-12 09:46:25 -08:00
..
CMakeLists.txt [HLO] Delete LHLO memref cast ops and migrate to STD ones. 2020-11-04 09:26:34 -08:00
chlo_legalize_to_hlo.cc Move unranked chlo lowering to transform_unranked_hlo. 2020-10-30 02:56:44 -07:00
chlo_legalize_to_hlo_pass.cc Integrate LLVM at llvm/llvm-project@0fc1aa22ee 2020-10-27 06:56:16 -07:00
chlo_legalize_to_hlo_patterns.td Add compare_type optional attribute to CompareOp in HLO dialects 2020-10-26 12:58:29 -07:00
hlo_legalize_to_lhlo.cc [MLIR:HLO] Extend CustomCall to support multiple outputs. 2020-11-12 09:46:25 -08:00
legalize_control_flow.cc Remove unused `using` statements. 2020-10-05 15:53:59 -07:00
legalize_gather_to_torch_index_select.cc Integrate LLVM at llvm/llvm-project@0fc1aa22ee 2020-10-27 06:56:16 -07:00
legalize_to_linalg.cc [MLIR] Add a lmhlo.reduce -> linalg.generic converter 2020-10-28 16:38:19 -07:00
legalize_to_standard.cc Integrate LLVM at llvm/llvm-project@0fc1aa22ee 2020-10-27 06:56:16 -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 Integrate LLVM at llvm/llvm-project@0fc1aa22ee 2020-10-27 06:56:16 -07:00
lhlo_fuse_linalg.cc Drop OperationFolder usage with Linalg fusion. 2020-10-29 06:10:14 -07:00
lhlo_legalize_to_affine.cc Integrate LLVM at llvm/llvm-project@0fc1aa22ee 2020-10-27 06:56:16 -07:00
lhlo_legalize_to_gpu.cc Integrate LLVM at llvm/llvm-project@0fc1aa22ee 2020-10-27 06:56:16 -07:00
lhlo_legalize_to_parallel_loops.cc Integrate LLVM at llvm/llvm-project@0fc1aa22ee 2020-10-27 06:56:16 -07:00
lower_complex.cc Integrate LLVM at llvm/llvm-project@0fc1aa22ee 2020-10-27 06:56:16 -07:00
lower_complex_patterns.td Removed Op(Complex, Real) lowering to address complex type inference issue 2020-11-06 15:24:23 -08:00
lower_general_dot.cc Integrate LLVM at llvm/llvm-project@0fc1aa22ee 2020-10-27 06:56:16 -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 Integrate LLVM at llvm/llvm-project@0fc1aa22ee 2020-10-27 06:56:16 -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 Integrate LLVM at llvm/llvm-project@0fc1aa22ee 2020-10-27 06:56:16 -07:00
sink_constants_to_control_flow.cc Add a header for table-gen generated pass for MHLO and use it in SinkConstantsToControlFlowPass 2020-09-21 18:01:57 -07:00
test_infer_shaped_type_pass.cc Integrate LLVM at llvm/llvm-project@0fc1aa22ee 2020-10-27 06:56:16 -07:00
transform_unranked_hlo.cc Handle rank 1 broadcasts in unranked kernel lowering. 2020-11-06 07:22:43 -08: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 Integrate LLVM at llvm/llvm-project@0fc1aa22ee 2020-10-27 06:56:16 -07:00