From 5ba07117443a2ca63cf38f099ef3e8a4e0de47eb Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 8 Apr 2020 17:59:09 -0700 Subject: [PATCH] Update MLIR users after replacing cl::opts with pass options PiperOrigin-RevId: 305592544 Change-Id: Ifa32c79945a9e106a5720bc9132cbd17cd371b2a --- tensorflow/compiler/mlir/lite/tests/inlining.mlir | 2 +- tensorflow/compiler/mlir/lite/tests/legalize-tf-while.mlir | 2 +- tensorflow/compiler/mlir/tensorflow/tests/inlining.mlir | 2 +- tensorflow/compiler/mlir/xla/transforms/lhlo_fuse_linalg.cc | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tensorflow/compiler/mlir/lite/tests/inlining.mlir b/tensorflow/compiler/mlir/lite/tests/inlining.mlir index ff7921b8aca..8f19e9983b5 100644 --- a/tensorflow/compiler/mlir/lite/tests/inlining.mlir +++ b/tensorflow/compiler/mlir/lite/tests/inlining.mlir @@ -1,4 +1,4 @@ -// RUN: tf-opt %s -inline -mlir-disable-inline-simplify | FileCheck %s --dump-input=fail +// RUN: tf-opt %s -inline="disable-simplify" | FileCheck %s --dump-input=fail // Inline a function that contains only tfl ops. func @func_with_tfl_ops(%arg0 : tensor<2xi32>) -> tensor<2xi32> { diff --git a/tensorflow/compiler/mlir/lite/tests/legalize-tf-while.mlir b/tensorflow/compiler/mlir/lite/tests/legalize-tf-while.mlir index 50ea5c1da41..c66e6d0145c 100644 --- a/tensorflow/compiler/mlir/lite/tests/legalize-tf-while.mlir +++ b/tensorflow/compiler/mlir/lite/tests/legalize-tf-while.mlir @@ -1,5 +1,5 @@ // RUN: tf-opt --tfl-legalize-tf-while %s -o - | FileCheck %s --dump-input-on-failure -// RUN: tf-opt --tfl-legalize-tf-while %s -o - --tfl-legalize-tf-while --inline --mlir-disable-inline-simplify | FileCheck %s --dump-input-on-failure --check-prefix=INLINE +// RUN: tf-opt --tfl-legalize-tf-while %s -o - --tfl-legalize-tf-while --inline="disable-simplify" | FileCheck %s --dump-input-on-failure --check-prefix=INLINE // RUN: tf-opt --tfl-legalize-tf-while %s -o - --tfl-legalize-tf-while --inline | FileCheck %s --dump-input-on-failure --check-prefix=CANON func @while_main(%arg0: tensor) -> (tensor, tensor<256x256xf32>, tensor) attributes {tf.entry_function = {inputs = "input", outputs = "Identity,Identity_1,Identity_2"}} { diff --git a/tensorflow/compiler/mlir/tensorflow/tests/inlining.mlir b/tensorflow/compiler/mlir/tensorflow/tests/inlining.mlir index 4dce57fbd10..77e53ec041e 100644 --- a/tensorflow/compiler/mlir/tensorflow/tests/inlining.mlir +++ b/tensorflow/compiler/mlir/tensorflow/tests/inlining.mlir @@ -1,4 +1,4 @@ -// RUN: tf-opt %s -inline -mlir-disable-inline-simplify | FileCheck %s --dump-input=fail +// RUN: tf-opt %s -inline="disable-simplify" | FileCheck %s --dump-input=fail // Test that simple TF operations can be inlined. diff --git a/tensorflow/compiler/mlir/xla/transforms/lhlo_fuse_linalg.cc b/tensorflow/compiler/mlir/xla/transforms/lhlo_fuse_linalg.cc index 256a6e531ab..bdee1b77cff 100644 --- a/tensorflow/compiler/mlir/xla/transforms/lhlo_fuse_linalg.cc +++ b/tensorflow/compiler/mlir/xla/transforms/lhlo_fuse_linalg.cc @@ -35,7 +35,7 @@ class LhloFuseLinalg : public PassWrapper { LhloFuseLinalg() = default; LhloFuseLinalg(const LhloFuseLinalg&) {} LhloFuseLinalg(bool use_parallel_loops, llvm::ArrayRef tile_sizes) { - tile_sizes_->assign(tile_sizes.begin(), tile_sizes.end()); + tile_sizes_ = tile_sizes; use_parallel_loops_.setValue(use_parallel_loops); }