mlir-hlo/lib/Dialect/mhlo/IR
Smit Hinsu 9bd1995f90 Legalize XlaReplicaId to HLO replica-id op
Also, define shape inference function for HLO replica-id op.

PiperOrigin-RevId: 345714342
2020-12-04 11:04:40 -08:00
..
CMakeLists.txt [MLIR:HLO] Generate enum decls for HLO and LHLO GPU dialects. 2020-12-02 11:39:23 -08: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 Add compare_type optional attribute to CompareOp in HLO dialects 2020-10-26 12:58:29 -07:00
hlo_ops.cc Legalize XlaReplicaId to HLO replica-id op 2020-12-04 11:04:40 -08: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 Make LMHLO's Dot have the same power as MHLO's DotGeneral. 2020-10-15 15:09:06 -07:00
hlo_patterns.td Add canonicalizer for Reshape(Broadcast(X)) pattern when it is an identity sequence 2020-11-19 02:32:45 -08: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 Make LMHLO's Dot have the same power as MHLO's DotGeneral. 2020-10-15 15:09:06 -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 [HLO] Delete LHLO memref cast ops and migrate to STD ones. 2020-11-04 09:26:34 -08:00
lhlo_ops_structs.cc Make LMHLO's Dot have the same power as MHLO's DotGeneral. 2020-10-15 15:09:06 -07:00
mhlo_canonicalize.td Add a canonicalization pattern to remove redundant dynamic_reshapes. 2020-11-27 04:46:50 -08:00