No public change.

cezheng · tensorflower-gardener · commit 2e9176558f27 · 2022-03-21T19:58:20.000-07:00
PiperOrigin-RevId: 436368495
diff --git a/tensorflow_estimator/python/estimator/tpu/BUILD b/tensorflow_estimator/python/estimator/tpu/BUILD
@@ -177,7 +177,6 @@ tpu_py_test(
     size = "medium",
     srcs = ["tpu_estimator_input_v2_test.py"],
     disable_experimental = True,
-    disable_tfrt = True,  # TODO(b/198521331)
     malloc = "//third_party/tcmalloc:tcmalloc_or_debug",
     srcs_version = "PY3",
     deps = [
