aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/estimator/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/python/estimator/BUILD')
-rw-r--r--tensorflow/python/estimator/BUILD121
1 files changed, 119 insertions, 2 deletions
diff --git a/tensorflow/python/estimator/BUILD b/tensorflow/python/estimator/BUILD
index e2501c0579..1404afa6ae 100644
--- a/tensorflow/python/estimator/BUILD
+++ b/tensorflow/python/estimator/BUILD
@@ -8,10 +8,10 @@ package(
],
)
-load("//tensorflow:tensorflow.bzl", "py_test")
-
licenses(["notice"]) # Apache 2.0
+load("//tensorflow:tensorflow.bzl", "py_test")
+
py_library(
name = "estimator_py",
srcs_version = "PY2AND3",
@@ -19,11 +19,24 @@ py_library(
":checkpoint_utils",
":estimator",
":export",
+ ":inputs",
":model_fn",
":run_config",
],
)
+filegroup(
+ name = "all_files",
+ srcs = glob(
+ ["**/*"],
+ exclude = [
+ "**/METADATA",
+ "**/OWNERS",
+ ],
+ ),
+ visibility = ["//tensorflow:__subpackages__"],
+)
+
py_library(
name = "checkpoint_utils",
srcs = ["checkpoint_utils.py"],
@@ -130,3 +143,107 @@ py_test(
"//tensorflow/python:client_testlib",
],
)
+
+py_library(
+ name = "inputs",
+ srcs = ["inputs/__init__.py"],
+ srcs_version = "PY2AND3",
+ deps = [
+ ":numpy_io",
+ ":pandas_import",
+ ":pandas_io",
+ ],
+)
+
+py_library(
+ name = "numpy_io",
+ srcs = ["inputs/numpy_io.py"],
+ srcs_version = "PY2AND3",
+ deps = [
+ ":inputs_queues",
+ ],
+)
+
+py_test(
+ name = "numpy_io_test",
+ size = "small",
+ srcs = ["inputs/numpy_io_test.py"],
+ srcs_version = "PY2AND3",
+ deps = [
+ ":numpy_io",
+ "//tensorflow/python:client_testlib",
+ "//tensorflow/python:errors",
+ "//tensorflow/python:framework_test_lib",
+ ],
+)
+
+py_library(
+ name = "pandas_import",
+ srcs = ["inputs/pandas_import.py"],
+ srcs_version = "PY2AND3",
+)
+
+py_library(
+ name = "pandas_io",
+ srcs = ["inputs/pandas_io.py"],
+ srcs_version = "PY2AND3",
+ deps = [
+ ":inputs_queues",
+ ":pandas_import",
+ ],
+)
+
+py_test(
+ name = "pandas_io_test",
+ size = "small",
+ srcs = ["inputs/pandas_io_test.py"],
+ srcs_version = "PY2AND3",
+ deps = [
+ ":pandas_io",
+ "//tensorflow/python:client_testlib",
+ "//tensorflow/python:errors",
+ "//tensorflow/python:framework_test_lib",
+ "//tensorflow/python:training",
+ ],
+)
+
+py_library(
+ name = "inputs_queues",
+ srcs = [
+ "inputs/queues/__init__.py",
+ "inputs/queues/feeding_functions.py",
+ "inputs/queues/feeding_queue_runner.py",
+ ],
+ srcs_version = "PY2AND3",
+ deps = [
+ ":pandas_import",
+ "//tensorflow/python:training",
+ ],
+)
+
+py_test(
+ name = "feeding_functions_test",
+ size = "small",
+ srcs = [
+ "inputs/queues/feeding_functions_test.py",
+ ],
+ srcs_version = "PY2AND3",
+ deps = [
+ ":inputs_queues",
+ "//tensorflow/python:client_testlib",
+ "//tensorflow/python:framework_test_lib",
+ ],
+)
+
+py_test(
+ name = "feeding_queue_runner_test",
+ size = "small",
+ srcs = ["inputs/queues/feeding_queue_runner_test.py"],
+ srcs_version = "PY2AND3",
+ deps = [
+ ":inputs_queues",
+ "//tensorflow/python:client_testlib",
+ "//tensorflow/python:errors",
+ "//tensorflow/python:framework_test_lib",
+ ],
+)