Create separate keras/estimator build
PiperOrigin-RevId: 375182231
Change-Id: I9e0c9aa2a67731183e51c17f918a15114ac5d080
diff --git a/tensorflow/python/keras/BUILD b/tensorflow/python/keras/BUILD
index 1eea171..6b1b00e 100755
--- a/tensorflow/python/keras/BUILD
+++ b/tensorflow/python/keras/BUILD
@@ -26,7 +26,6 @@
name = "keras",
srcs = [
"__init__.py",
- "estimator/__init__.py",
"keras_parameterized.py",
],
srcs_version = "PY3",
@@ -35,10 +34,10 @@
":engine",
":testing_utils", # TODO(scottzhu): Stop exporting the test_utils after removing all the callers.
"//tensorflow/python:training",
- "//tensorflow/python/eager:monitoring",
"//tensorflow/python/keras/applications",
"//tensorflow/python/keras/datasets",
"//tensorflow/python/keras/distribute",
+ "//tensorflow/python/keras/estimator",
"//tensorflow/python/keras/feature_column",
"//tensorflow/python/keras/layers",
"//tensorflow/python/keras/mixed_precision:mixed_precision_experimental",
diff --git a/tensorflow/python/keras/estimator/BUILD b/tensorflow/python/keras/estimator/BUILD
new file mode 100644
index 0000000..bac7808
--- /dev/null
+++ b/tensorflow/python/keras/estimator/BUILD
@@ -0,0 +1,21 @@
+# Description:
+# Contains Keras models to Estimator converter
+
+package(
+ default_visibility = [
+ "//tensorflow/python/keras:__subpackages__",
+ ],
+ licenses = ["notice"],
+)
+
+py_library(
+ name = "estimator",
+ srcs = [
+ "__init__.py",
+ ],
+ srcs_version = "PY3",
+ deps = [
+ "//tensorflow/python:util",
+ "//tensorflow/python/eager:monitoring",
+ ],
+)
