From 25a02eadb25f6ecbe48a166c204060f918648689 Mon Sep 17 00:00:00 2001
From: Nupur Garg <nupurgarg@google.com>
Date: Fri, 2 Aug 2019 14:12:09 -0700
Subject: [PATCH] Do not build with TFLite MLIR converter by default.

PiperOrigin-RevId: 261388784
---
 tensorflow/lite/python/BUILD | 1 +
 tensorflow/tensorflow.bzl    | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/tensorflow/lite/python/BUILD b/tensorflow/lite/python/BUILD
index ca005465212..9316da8e94c 100644
--- a/tensorflow/lite/python/BUILD
+++ b/tensorflow/lite/python/BUILD
@@ -141,6 +141,7 @@ py_test(
     srcs = ["lite_mlir_test.py"],
     srcs_version = "PY2AND3",
     tags = [
+        "no_oss",
         "no_windows",
     ],
     deps = [
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl
index 7693aaaeb2e..ce60a18b872 100644
--- a/tensorflow/tensorflow.bzl
+++ b/tensorflow/tensorflow.bzl
@@ -2525,7 +2525,7 @@ def if_mlir(if_true, if_false = []):
 
 # TODO(b/138724071): Remove when build is stable.
 def if_mlir_tflite(if_true, if_false = []):
-    return if_true  # Internally we always build with MLIR.
+    return if_mlir(if_true, if_false)
 
 def tfcompile_extra_flags():
     return ""