mlir-hlo/lib/Dialect/mhlo/IR
Wenyi Zhao 88cc0c6c46 PR #50271: [MLIR][DISC] Bufferize GatherOp and DynamicGatherOp
Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/50271

support hlo-to-lhlo conversion for GatherOp and DynamicGatherOp
Copybara import of the project:

--
117a1b1bcaac7ecc5224b02863eede5c1b9618fe by Wenyi Zhao <reyizero@gmail.com>:

[MLIR][DISC] Bufferize GatherOp and DynamicGatherOp

PiperOrigin-RevId: 379801972
2021-06-16 13:47:56 -07:00
..
CMakeLists.txt PR #50191: [MLIR][DISC] Add RAL (Runtime abstraction layer) Dialect 2021-06-14 11:27:43 -07:00
chlo_canonicalize.td Add chlo.constant_like op which splats a constant to shape of operand 2020-08-11 14:54:48 -07:00
chlo_ops.cc PR #49454: [MLIR][DISC] Upgrade to use the new `reifyReturnTypeShapes` interface. 2021-05-24 10:11:55 -07:00
disc_ral_ops.cc PR #50191: [MLIR][DISC] Add RAL (Runtime abstraction layer) Dialect 2021-06-14 11:27:43 -07:00
hlo_ops.cc PR #50271: [MLIR][DISC] Bufferize GatherOp and DynamicGatherOp 2021-06-16 13:47:56 -07:00
hlo_ops_base_enums.cc [MLIR:HLO] Generate enum decls for HLO and LHLO GPU dialects. 2020-12-02 11:39:23 -08:00
hlo_ops_base_structs.cc [HLO] Add custom print/parse for convolution dimension numbers (in LMHLO) 2021-05-12 08:52:46 -07:00
hlo_ops_common.cc [HLO] Add AllReduceScatter to MHLO and LMHLO dialects. 2021-06-14 09:37:07 -07:00
hlo_patterns.td Restrict canonicalization to avoid changing type 2021-03-16 16:54:05 -07:00
infer_fusibility_op_interface.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
init.cc Add GPU specific LMHLO level ops 2020-10-14 11:23:55 -07:00
lhlo_gpu_ops.cc [XLA:GPU] Add AllReduce{Start,Done} to MLIR LHLO dialect. 2021-06-15 03:55:19 -07:00
lhlo_gpu_ops_enums.cc [MLIR:HLO] Generate enum decls for HLO and LHLO GPU dialects. 2020-12-02 11:39:23 -08:00
lhlo_gpu_ops_structs.cc Make LMHLO's Dot have the same power as MHLO's DotGeneral. 2020-10-15 15:09:06 -07:00
lhlo_ops.cc [XLA:GPU] Add AllReduce{Start,Done} to MLIR LHLO dialect. 2021-06-15 03:55:19 -07:00
lhlo_ops_structs.cc [MLIR:LHLO] Add optional call target arg mapping to LMHLO CustomCall operations. 2021-02-22 08:43:00 -08:00
mhlo_canonicalize.td PR #49919: [MLIR][DISC] pattern conversion from tf2mhlo: ConvertUnpackOpDynamic, ConvertSignOpDynamic, ConvertSigmoidGradOpDynamic 2021-06-15 10:33:32 -07:00