diff --git a/tensorflow/compiler/mlir/tensorflow/tests/op_fusion.mlir b/tensorflow/compiler/mlir/tensorflow/tests/op_fusion.mlir index 4688d2ee712..77ce19da8a4 100644 --- a/tensorflow/compiler/mlir/tensorflow/tests/op_fusion.mlir +++ b/tensorflow/compiler/mlir/tensorflow/tests/op_fusion.mlir @@ -1,4 +1,4 @@ -// RUN: tf-opt %s -op-fusion | FileCheck %s --dump-input-on-failure +// RUN: tf-opt %s -tf-op-fusion | FileCheck %s --dump-input-on-failure //===----------------------------------------------------------------------===// // Conv2D + BiasAdd + fusions. diff --git a/tensorflow/compiler/mlir/tensorflow/tests/readonly_references_to_resources.mlir b/tensorflow/compiler/mlir/tensorflow/tests/readonly_references_to_resources.mlir index 2970e31c3c9..fe3234bcc4e 100644 --- a/tensorflow/compiler/mlir/tensorflow/tests/readonly_references_to_resources.mlir +++ b/tensorflow/compiler/mlir/tensorflow/tests/readonly_references_to_resources.mlir @@ -1,4 +1,4 @@ -// RUN: tf-opt -verify-diagnostics -readonly-references-to-resources -split-input-file %s | FileCheck %s --dump-input=fail +// RUN: tf-opt -verify-diagnostics -tf-readonly-references-to-resources -split-input-file %s | FileCheck %s --dump-input=fail // Test case: Basic converting. diff --git a/tensorflow/compiler/mlir/tensorflow/transforms/op_fusion.cc b/tensorflow/compiler/mlir/tensorflow/transforms/op_fusion.cc index 202783d1cc8..0349a5b26ba 100644 --- a/tensorflow/compiler/mlir/tensorflow/transforms/op_fusion.cc +++ b/tensorflow/compiler/mlir/tensorflow/transforms/op_fusion.cc @@ -166,7 +166,7 @@ std::unique_ptr> CreateOpFusionPass() { } static PassRegistration pass( - "op-fusion", + "tf-op-fusion", "Replaces commonly occurring subgraphs with optimized fused kernels"); } // namespace TF diff --git a/tensorflow/compiler/mlir/tensorflow/transforms/readonly_references_to_resources.cc b/tensorflow/compiler/mlir/tensorflow/transforms/readonly_references_to_resources.cc index a80b84ddeda..5fc35361bca 100644 --- a/tensorflow/compiler/mlir/tensorflow/transforms/readonly_references_to_resources.cc +++ b/tensorflow/compiler/mlir/tensorflow/transforms/readonly_references_to_resources.cc @@ -171,7 +171,7 @@ CreateConvertReadonlyReferenceVariablesToResourceVariablesPass() { static PassRegistration< ConvertReadonlyReferenceVariablesToResourceVariablesPass> - pass("readonly-references-to-resources", + pass("tf-readonly-references-to-resources", "Convert readonly reference variables to resource variables."); } // namespace TF