experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[MLIR:HLO] Change LHLO Outfeed to support multiple inputs.

Browse Source 
PiperOrigin-RevId: 351447034
Change-Id: Id67e49ae2bf39a53b8e3771a719f22b081266aeb
This commit is contained in:
Rahul Joshi 2021-01-12 14:12:43 -08:00 committed by TensorFlower Gardener
parent 868d16e045
commit 5edd67fd01
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -584,7 +584,7 @@ def LHLO_InfeedOp: LHLO_Op<"infeed", []>, BASE_HLO_InfeedOp {
def LHLO_OutfeedOp: LHLO_Op<"outfeed", []> {
let arguments = (ins
Arg<LHLO_Buffer, "", [MemRead]>:$operand,
Arg<Variadic<LHLO_Buffer>, "", [MemRead]>:$operands,
DefaultValuedAttr<StrAttr, "">:$config
);
}