From a8200b181895f3ad427f6a692dc10014935305f3 Mon Sep 17 00:00:00 2001 From: Tzu-Wei Sung Date: Fri, 29 Jan 2021 18:22:43 -0800 Subject: [PATCH] Update test descriptions --- .../tests/region-control-flow-to-functional.mlir | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tensorflow/compiler/mlir/tensorflow/tests/region-control-flow-to-functional.mlir b/tensorflow/compiler/mlir/tensorflow/tests/region-control-flow-to-functional.mlir index 48f7786c870..6a5d702accd 100644 --- a/tensorflow/compiler/mlir/tensorflow/tests/region-control-flow-to-functional.mlir +++ b/tensorflow/compiler/mlir/tensorflow/tests/region-control-flow-to-functional.mlir @@ -168,11 +168,11 @@ func @testIf2Result(%arg0: tensor, %arg1: tensor<2xf32>) -> tensor<2xf32> { // ----- -// Do not skip incompatible cast for trivial transform. +// Do not skip extern incompatible cast for trivial transform. func private @testIf1Then(tensor<*xf32>) -> tensor<*xf32> func private @testIf1Else(tensor<*xf32>) -> tensor<*xf32> -func @testIfIncompatibleCastTrivialTransform(%arg0: tensor, %arg1: tensor<2xi64>) -> tensor<2xf32> { +func @testIfExternIncompatibleCastTrivialTransform(%arg0: tensor, %arg1: tensor<2xi64>) -> tensor<2xf32> { // CHECK: %[[CAST:.*]] = "tf.Cast"(%arg1) {Truncate = false} : (tensor<2xi64>) -> tensor<*xf32> // CHECK: "tf.If"(%arg0, %[[CAST]]) {else_branch = @testIf1Else, {{.+}} then_branch = @testIf1Then} %1 = "tf.Cast"(%arg1) {Truncate = false} : (tensor<2xi64>) -> tensor<*xf32> @@ -188,7 +188,7 @@ func @testIfIncompatibleCastTrivialTransform(%arg0: tensor, %arg1: tensor<2x // ----- -// Do not skip incompatible cast for non-trivial transform. +// Do not skip incompatible cast for trivial transform. // CHECK: func private @tf.IfRegion_else(%arg0: tensor<2xi64>) -> tensor<*xf32> // CHECK-NEXT: "tf.Cast" @@ -196,7 +196,7 @@ func @testIfIncompatibleCastTrivialTransform(%arg0: tensor, %arg1: tensor<2x // CHECK-NEXT: "tf.Cast" func private @testIf1Then(tensor<*xf32>) -> tensor<*xf32> func private @testIf1Else(tensor<*xf32>) -> tensor<*xf32> -func @testIfIncompatibleCastNonTrivialTransform(%arg0: tensor, %arg1: tensor<2xi64>) -> tensor<2xf32> { +func @testIfIncompatibleCastTrivialTransform(%arg0: tensor, %arg1: tensor<2xi64>) -> tensor<2xf32> { // CHECK: "tf.If"(%arg0, %arg1) {else_branch = @tf.IfRegion_else{{.+}}then_branch = @tf.IfRegion_then} %0 = "tf.IfRegion"(%arg0) ( { %1 = "tf.Cast"(%arg1) {Truncate = false} : (tensor<2xi64>) -> tensor<*xf32>