diff --git a/tensorflow/compiler/mlir/lite/ir/tfl_ops.cc b/tensorflow/compiler/mlir/lite/ir/tfl_ops.cc index 318ca36a96b..7835191037c 100644 --- a/tensorflow/compiler/mlir/lite/ir/tfl_ops.cc +++ b/tensorflow/compiler/mlir/lite/ir/tfl_ops.cc @@ -2089,8 +2089,7 @@ struct WhileResultOperandsMatchAndImplicitCapture Operation *op = while_op.getOperation(); Operation *new_op = rewriter.insert( Operation::create(op->getLoc(), op->getName(), types, new_operands, - op->getAttrs(), {}, /*numRegions=*/2, - /*resizableOperandList=*/true)); + op->getAttrs(), {}, /*numRegions=*/2)); for (int i = 0; i < 2; ++i) new_op->getRegion(i).takeBody(op->getRegion(i)); int new_index = 0; diff --git a/tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc b/tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc index a7f2a625e65..707f4aba881 100644 --- a/tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc +++ b/tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc @@ -228,8 +228,7 @@ void WhileOutlinePass::OutlineWhile(WhileOp while_op) { Operation* new_op = OpBuilder(op).insert(Operation::create( op->getLoc(), op->getName(), new_types, operands, op->getAttrs(), - /*successors=*/{}, /*numRegions=*/2, - /*resizableOperandList=*/true)); + /*successors=*/{}, /*numRegions=*/2)); for (int i = 0; i < 2; ++i) new_op->getRegion(i).takeBody(op->getRegion(i)); op->replaceAllUsesWith(new_op->getResults().take_front(op->getNumResults())); op->erase(); diff --git a/third_party/mlir/BUILD b/third_party/mlir/BUILD index cd3f384ea42..71344a0cd00 100644 --- a/third_party/mlir/BUILD +++ b/third_party/mlir/BUILD @@ -654,6 +654,7 @@ cc_library( deps = [ ":CallOpInterfaces", ":CommonFolders", + ":Dialect", ":IR", ":InferTypeOpInterface", ":ShapeOpsIncGen",