From 0d61fc79f51fa79b7836eb5647beb537abb73707 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Fri, 3 Apr 2020 21:48:38 -0700
Subject: [PATCH] Update for llvm 71305033d115

PiperOrigin-RevId: 304748166
Change-Id: If3744b05c773cd5343e1328a01a6716a050a8674
---
 .../xla/transforms/lhlo_legalize_to_parallel_loops.cc  | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/tensorflow/compiler/mlir/xla/transforms/lhlo_legalize_to_parallel_loops.cc b/tensorflow/compiler/mlir/xla/transforms/lhlo_legalize_to_parallel_loops.cc
index f2fc32f63ff..2781b955327 100644
--- a/tensorflow/compiler/mlir/xla/transforms/lhlo_legalize_to_parallel_loops.cc
+++ b/tensorflow/compiler/mlir/xla/transforms/lhlo_legalize_to_parallel_loops.cc
@@ -79,8 +79,9 @@ struct MappedIvs {
   SmallVector<Value, 2> ivs;
 };
 
-MappedIvs MapWindowIvsToInput(ReduceWindowOp op, ArrayRef<Value> ivs,
-                              ArrayRef<Value> window_ivs, OpBuilder* b) {
+MappedIvs MapWindowIvsToInput(ReduceWindowOp op, ArrayRef<BlockArgument> ivs,
+                              ArrayRef<BlockArgument> window_ivs,
+                              OpBuilder* b) {
   MappedIvs mapped_ivs;
 
   if (!op.window_strides().hasValue()) {
@@ -418,9 +419,8 @@ class ReduceWindowOpConverter
     auto xla_operand_type = xla_operand.getType().cast<MemRefType>();
 
     MappedIvs mapped_ivs = MapWindowIvsToInput(
-        xla_reduce_window_op,
-        SmallVector<Value, 2>{output_loop.getInductionVars()},
-        SmallVector<Value, 2>{window_loop.getInductionVars()}, rewriter);
+        xla_reduce_window_op, output_loop.getInductionVars(),
+        window_loop.getInductionVars(), rewriter);
 
     auto elem_or_init = rewriter->create<loop::IfOp>(
         loc, xla_operand_type.getElementType(), mapped_ivs.in_bounds,