Skip to content

Commit e1c4c45

Browse files
jurahultensorflower-gardener
authored andcommitted
[MLIR:HLO] Rename LMHLO Infeed and Outfeed to InfeedOp and OutfeedOp.
- To be consistent with other operation names. PiperOrigin-RevId: 348824379 Change-Id: Ie8f19bb6cba3147b4ce527c3fb6d84c76f472af7
1 parent a5bc02b commit e1c4c45

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -575,14 +575,14 @@ def LHLO_CholeskyOp: LHLO_Op<"cholesky", [SameOperandsElementType]>, BASE_HLO_Ch
575575
);
576576
}
577577

578-
def LHLO_Infeed: LHLO_Op<"infeed", []>, BASE_HLO_InfeedOp {
578+
def LHLO_InfeedOp: LHLO_Op<"infeed", []>, BASE_HLO_InfeedOp {
579579
let arguments = (ins
580580
Arg<LHLO_Buffer, "", [MemWrite]>:$output,
581581
DefaultValuedAttr<StrAttr, "">:$config
582582
);
583583
}
584584

585-
def LHLO_Outfeed: LHLO_Op<"outfeed", []> {
585+
def LHLO_OutfeedOp: LHLO_Op<"outfeed", []> {
586586
let arguments = (ins
587587
Arg<LHLO_Buffer, "", [MemRead]>:$operand,
588588
DefaultValuedAttr<StrAttr, "">:$config

tensorflow/compiler/mlir/xla/hlo_utils.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -238,9 +238,9 @@ StatusOr<::xla::HloOpcode> MhloToHloOpcode(mlir::Operation* op) {
238238
return xla::HloOpcode::kSubtract;
239239
} else if (isa<mlir::mhlo::XorOp, mlir::lmhlo::XorOp>(op)) {
240240
return xla::HloOpcode::kXor;
241-
} else if (isa<mlir::mhlo::InfeedOp, mlir::lmhlo::Infeed>(op)) {
241+
} else if (isa<mlir::mhlo::InfeedOp, mlir::lmhlo::InfeedOp>(op)) {
242242
return xla::HloOpcode::kInfeed;
243-
} else if (isa<mlir::mhlo::OutfeedOp, mlir::lmhlo::Outfeed>(op)) {
243+
} else if (isa<mlir::mhlo::OutfeedOp, mlir::lmhlo::OutfeedOp>(op)) {
244244
return xla::HloOpcode::kOutfeed;
245245
} else if (isa<mlir::mhlo::SendOp>(op)) {
246246
return xla::HloOpcode::kSend;

0 commit comments

Comments
 (0)