From 1acf61f0281ff2803c64c2e04fb95a2f2b78e5ab Mon Sep 17 00:00:00 2001 From: Andy Ly Date: Mon, 8 Jun 2020 14:35:55 -0700 Subject: [PATCH] Prefix OpFusion and ConvertReadonlyReferenceVariablesToResourceVariablesPass pass flag with 'tf'. (NFC) Other passes have the 'tf' prefix for their flags, so this is to have flags be more consistent with one another, under the tensorflow directory. PiperOrigin-RevId: 315355169 Change-Id: I42258041c5b455a140f5e338f146b305a16f7a83 --- tensorflow/compiler/mlir/tensorflow/tests/op_fusion.mlir | 2 +- .../mlir/tensorflow/tests/readonly_references_to_resources.mlir | 2 +- tensorflow/compiler/mlir/tensorflow/transforms/op_fusion.cc | 2 +- .../tensorflow/transforms/readonly_references_to_resources.cc | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) 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