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

Update MLIR users after replacing cl::opts with pass options 

PiperOrigin-RevId: 305592544
Change-Id: Ifa32c79945a9e106a5720bc9132cbd17cd371b2a
This commit is contained in:
A. Unique TensorFlower 2020-04-08 17:59:09 -07:00 committed by TensorFlower Gardener
parent 2c47fddbb8
commit 5ba0711744
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/mlir/lite/tests/inlining.mlir
View File

@ -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. // Inline a function that contains only tfl ops.
func @func_with_tfl_ops(%arg0 : tensor<2xi32>) -> tensor<2xi32> { func @func_with_tfl_ops(%arg0 : tensor<2xi32>) -> tensor<2xi32> {

2
tensorflow/compiler/mlir/lite/tests/legalize-tf-while.mlir
View File

@ -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 - | 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 // 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<?x256x256xf32>) -> (tensor<i32>, tensor<256x256xf32>, tensor<?x256x256xf32>) attributes {tf.entry_function = {inputs = "input", outputs = "Identity,Identity_1,Identity_2"}} { func @while_main(%arg0: tensor<?x256x256xf32>) -> (tensor<i32>, tensor<256x256xf32>, tensor<?x256x256xf32>) attributes {tf.entry_function = {inputs = "input", outputs = "Identity,Identity_1,Identity_2"}} {

2
tensorflow/compiler/mlir/tensorflow/tests/inlining.mlir
View File

@ -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. // Test that simple TF operations can be inlined.

2
tensorflow/compiler/mlir/xla/transforms/lhlo_fuse_linalg.cc
View File

@ -35,7 +35,7 @@ class LhloFuseLinalg : public PassWrapper<LhloFuseLinalg, FunctionPass> {
LhloFuseLinalg() = default; LhloFuseLinalg() = default;
LhloFuseLinalg(const LhloFuseLinalg&) {} LhloFuseLinalg(const LhloFuseLinalg&) {}
LhloFuseLinalg(bool use_parallel_loops, llvm::ArrayRef<unsigned> tile_sizes) { LhloFuseLinalg(bool use_parallel_loops, llvm::ArrayRef<unsigned> tile_sizes) {
tile_sizes_->assign(tile_sizes.begin(), tile_sizes.end()); tile_sizes_ = tile_sizes;
use_parallel_loops_.setValue(use_parallel_loops); use_parallel_loops_.setValue(use_parallel_loops);
} }