commit 5edd67fd01d41d77f8b26eeef8beb550243633fc
author Rahul Joshi <jurahul@google.com> Tue Jan 12 14:12:43 2021 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> Tue Jan 12 14:45:38 2021 -0800
tree b6c1adad7fda2a97f4fd875c61d6ce6ea22c09ee
parent 868d16e045071d7eff3bcf594a83dc3d5ab77bff

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

PiperOrigin-RevId: 351447034
Change-Id: Id67e49ae2bf39a53b8e3771a719f22b081266aeb

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

diff --git a/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td b/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td
index c549368..b5c2a73 100644
--- a/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td
+++ b/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td
@@ -584,7 +584,7 @@
 
 def LHLO_OutfeedOp: LHLO_Op<"outfeed", []> {
   let arguments = (ins
-    Arg<LHLO_Buffer, "", [MemRead]>:$operand,
+    Arg<Variadic<LHLO_Buffer>, "", [MemRead]>:$operands,
     DefaultValuedAttr<StrAttr, "">:$config
   );
 }