colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
mlir-hlo/lib/Dialect/mhlo/transforms
History
Hanhan Wang 90f0d7f935 Add support for lowering mhlo.conv to Linalg on tensors. 
This pattern only works for normal convolutions. It does not work for depthwise
convolutions. The Linalg conv ops are defined with static rank, so it only
supports 1d/2d/3d cases, which are the most typical cases.

This also refactors out the same check in lmhlo.conv lowering.

PiperOrigin-RevId: 359503527
 		2021-02-25 05:59:08 -08:00
..
CMakeLists.txt Integrate LLVM at llvm/llvm-project@2bfe27da17 2021-02-12 08:32:03 -08:00
chlo_legalize_to_hlo.cc [MLIR][HLO] Remove duplicate `PopulateTransformUnrankedHloPatterns` 2021-02-23 07:50:47 -08:00
chlo_legalize_to_hlo_pass.cc [MLIR][CHLO] Add legalization for `chlo.polygamma` to MHLO 2021-02-17 08:33:01 -08:00
chlo_legalize_to_hlo_patterns.td [MLIR][CHLO] Add `is_inf`, `is_pos_inf`, and `is_neg_inf` to CHLO dialect 2021-01-27 09:00:56 -08:00
hlo_legalize_to_lhlo.cc Lower Expm1 kernel to math.ExpM1. 2021-02-18 04:54:23 -08:00
legalize_control_flow.cc Fix MLIR include paths. 2020-12-17 00:56:04 -08:00
legalize_gather_to_torch_index_select.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08:00
legalize_to_linalg.cc Add support for lowering mhlo.conv to Linalg on tensors. 2021-02-25 05:59:08 -08:00
legalize_to_standard.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08: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@2bfe27da17 2021-02-12 08:32:03 -08:00
lhlo_fuse_linalg.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08:00
lhlo_legalize_to_affine.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08:00
lhlo_legalize_to_gpu.cc Integrate LLVM at llvm/llvm-project@91e7a17133 2021-02-04 13:42:31 -08:00
lhlo_legalize_to_parallel_loops.cc [mlir][NFC] Replace usages or mlir/IR/StandardTypes.h with mlir/IR/BuiltinTypes.h 2020-12-15 00:59:29 -08:00
lower_complex.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08: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@d1978fa4bf 2021-02-05 07:42:06 -08: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 Fix MLIR include paths. 2020-12-17 00:56:04 -08: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@d1978fa4bf 2021-02-05 07:42:06 -08:00
sink_constants_to_control_flow.cc Enable fallback legalization for MaxPoolGradGrad and MaxPool3DGradGrad ops 2020-11-13 13:49:05 -08:00
test_infer_shaped_type_pass.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08:00
transform_unranked_hlo.cc [MLIR][HLO] Remove duplicate `PopulateTransformUnrankedHloPatterns` 2021-02-23 07:50:47 -08:00
unfuse_batch_norm.cc Integrate LLVM at llvm/llvm-project@96ef4f307d 2021-01-20 07:09:47 -08:00
unfuse_batch_norm_pass.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08:00