colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[MLIR:HLO] Rename LMHLO Infeed and Outfeed to InfeedOp and OutfeedOp. 

- To be consistent with other operation names.

PiperOrigin-RevId: 348824379
This commit is contained in:
Rahul Joshi 2020-12-23 11:35:54 -08:00 committed by TensorFlow MLIR Team
parent 8252eafa99
commit bd8e768e23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td
View File

@ -575,14 +575,14 @@ def LHLO_CholeskyOp: LHLO_Op<"cholesky", [SameOperandsElementType]>, BASE_HLO_Ch
); );
} }
def LHLO_Infeed: LHLO_Op<"infeed", []>, BASE_HLO_InfeedOp { def LHLO_InfeedOp: LHLO_Op<"infeed", []>, BASE_HLO_InfeedOp {
let arguments = (ins let arguments = (ins
Arg<LHLO_Buffer, "", [MemWrite]>:$output, Arg<LHLO_Buffer, "", [MemWrite]>:$output,
DefaultValuedAttr<StrAttr, "">:$config DefaultValuedAttr<StrAttr, "">:$config
); );
} }
def LHLO_Outfeed: LHLO_Op<"outfeed", []> { def LHLO_OutfeedOp: LHLO_Op<"outfeed", []> {
let arguments = (ins let arguments = (ins
Arg<LHLO_Buffer, "", [MemRead]>:$operand, Arg<LHLO_Buffer, "", [MemRead]>:$operand,
DefaultValuedAttr<StrAttr, "">:$config DefaultValuedAttr<StrAttr, "">:$config