mlir-hlo/lib/Dialect/mhlo/transforms
Adrian Kuegel fa059259bc Add template for tf.Cast
Also generate the kernels for all types of casts between signed int and float types.
This requires some adaptations to our build macros so that we can also specify the
output type of a kernel.

PiperOrigin-RevId: 354067727
2021-01-27 04:49:55 -08:00
..
CMakeLists.txt Fix CMakeLists.txt 2021-01-21 12:42:35 -08:00
chlo_legalize_to_hlo.cc [MLIR][KernelGen] Add erfc kernel for f16 2021-01-22 03:38:30 -08:00
chlo_legalize_to_hlo_pass.cc Fix header include. 2020-12-24 01:40:31 -08:00
chlo_legalize_to_hlo_patterns.td Add chlo.acosh operation and associated lowerings. 2021-01-20 11:43:44 -08:00
hlo_legalize_to_lhlo.cc [MLIR] Add mhlo.logistic lowering to linalg 2021-01-22 03:03:16 -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 [mlir][NFC] Replace usages of Function.h and Module.h with BuiltinOps.h 2020-12-01 13:18:06 -08:00
legalize_to_linalg.cc [MLIR] Add mhlo.logistic lowering to linalg 2021-01-22 03:03:16 -08:00
legalize_to_standard.cc Integrate LLVM at llvm/llvm-project@1b97cdf885 2020-12-21 23:49:18 -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 [mlir][NFC] Replace usages of Function.h and Module.h with BuiltinOps.h 2020-12-01 13:18:06 -08:00
lhlo_fuse_linalg.cc Integrate LLVM at llvm/llvm-project@6f4d460762 2021-01-12 16:38:25 -08:00
lhlo_legalize_to_affine.cc [mlir][NFC] Replace usages or mlir/IR/StandardTypes.h with mlir/IR/BuiltinTypes.h 2020-12-15 00:59:29 -08:00
lhlo_legalize_to_gpu.cc [mlir][NFC] Replace usages or mlir/IR/StandardTypes.h with mlir/IR/BuiltinTypes.h 2020-12-15 00:59:29 -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@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 [mlir][NFC] Replace usages or mlir/IR/StandardTypes.h with mlir/IR/BuiltinTypes.h 2020-12-15 00:59:29 -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@0fc1aa22ee 2020-10-27 06:56:16 -07: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@0fc1aa22ee 2020-10-27 06:56:16 -07:00
transform_unranked_hlo.cc Add template for tf.Cast 2021-01-27 04:49:55 -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@0fc1aa22ee 2020-10-27 06:56:16 -07:00