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

added missing CHECK, whitespace cleanup

Browse Source 
PiperOrigin-RevId: 311816309
Change-Id: I4181bf3d82e82f4ee60f2a894f487c19522490d1
This commit is contained in:
Aart Bik 2020-05-15 15:52:55 -07:00 committed by TensorFlower Gardener
parent b6284742e4
commit ec52e0fcd3
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/tf_program/tests/mlir_gen_test.py
View File

@ -83,7 +83,7 @@ class MLIRGenTest(MLIRGenTestBase):
CHECK-LABEL: func @test_fn(%arg0: i1, %arg1: i1) -> i1
CHECK: %[[r0:[0-9]+]] = "tfp.And"(%arg0, %arg0, %arg1) : (i1, i1, i1) -> tensor<*xi1>
CHECK: %[[r1:[0-9]+]] = "tfp.Or"(%arg0, %arg1, %[[r0]]) : (i1, i1, tensor<*xi1>) -> tensor<*xi1>
return %[[r1]] : tensor<*xi1>
CHECK: return %[[r1]] : tensor<*xi1>
"""
self._check_code(mlir_code, exp_mlir_code)
@ -158,7 +158,7 @@ class MLIRGenTest(MLIRGenTestBase):
mlir_code = mlir_gen(test_fn)
exp_mlir_code = r"""
CHECK-LABEL: func @test_fn(%arg0: tensor<*xi32>) -> i32
CHECK: %[[r1:[0-9]+]] = "tf.Greater"(%arg0, %{{[0-9]+}}) : (tensor<*xi32>, tensor<i32>) -> tensor<*xi1>
CHECK-NEXT: %[[r2:[0-9]+]] = "tfp.If"(%[[r1]]) ( {
CHECK: return %{{[0-9]+}} : tensor<i32>
@ -222,7 +222,7 @@ class MLIRGenTest(MLIRGenTestBase):
CHECK: %[[r5:[0-9]+]] = "tf.Equal"(%arg0, %{{[0-9]+}}) {incompatible_shape_error = true} : (tensor<*xi32>, tensor<i32>) -> tensor<*xi1>
CHECK: %[[r7:[0-9]+]] = "tf.Equal"(%arg0, %{{[0-9]+}}) {incompatible_shape_error = true} : (tensor<*xi32>, tensor<i32>) -> tensor<*xi1>
CHECK: %[[r8:[0-9]+]] = "tfp.Or"(%[[r5]], %[[r7]]) : (tensor<*xi1>, tensor<*xi1>) -> tensor<*xi1>
CHECK: %[[r9:[0-9]+]]:4 = "tfp.If"(%[[r8]]) ( {
CHECK-NEXT: return %{{.*}}, %{{.*}}, %{{.*}}, %{{.*}} : tensor<{{(\*x)?}}i32>, tensor<{{(\*x)?}}i32>, tensor<{{(\*x)?}}i32>, tensor<{{(\*x)?}}i32>
CHECK-NEXT: }, {