From d71b3dfeca6e59775033a6f4744462772ad536ad Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Tue, 5 May 2020 16:12:11 -0700
Subject: [PATCH] Fix legalizeTF(). Currently definition of this function is
inside an anonymous namespace.
PiperOrigin-RevId: 310040842
Change-Id: Ia37716447a9cb307c26f8c2f6efaee4915e35c19
---
.../mlir/xla/transforms/legalize_tf.cc | 22 +++++++++----------
1 file changed, 11 insertions(+), 11 deletions(-)
diff --git a/tensorflow/compiler/mlir/xla/transforms/legalize_tf.cc b/tensorflow/compiler/mlir/xla/transforms/legalize_tf.cc
index a6a6829b109..fb03c9b82e5 100644
--- a/tensorflow/compiler/mlir/xla/transforms/legalize_tf.cc
+++ b/tensorflow/compiler/mlir/xla/transforms/legalize_tf.cc
@@ -4785,6 +4785,17 @@ class ConvertQrOp : public OpRewritePattern<TF::QrOp> {
}
};
+// Performs the lowering to XLA dialect.
+void LegalizeTF::runOnFunction() {
+ if (failed(legalizeTF(getFunction(), allow_partial_conversion_)))
+ signalPassFailure();
+}
+
+static PassRegistration<LegalizeTF> pass(
+ "xla-legalize-tf", "Legalize from TensorFlow to the XLA dialect");
+
+} // end namespace
+
#include "tensorflow/compiler/mlir/xla/transforms/generated_legalize_tf.inc"
LogicalResult legalizeTF(Operation *op, bool allow_partial_conversion) {
@@ -4836,17 +4847,6 @@ LogicalResult legalizeTF(Operation *op, bool allow_partial_conversion) {
return applyPartialConversion(op, target, patterns);
}
-/// Performs the lowering to XLA dialect.
-void LegalizeTF::runOnFunction() {
- if (failed(legalizeTF(getFunction(), allow_partial_conversion_)))
- signalPassFailure();
-}
-
-static PassRegistration<LegalizeTF> pass(
- "xla-legalize-tf", "Legalize from TensorFlow to the XLA dialect");
-
-} // end namespace
-
std::unique_ptr<OperationPass<FuncOp>> createLegalizeTFPass(
bool allow_partial_conversion) {
return std::make_unique<LegalizeTF>(allow_partial_conversion);