diff options
Diffstat (limited to 'tensorflow/python/estimator/BUILD')
-rw-r--r-- | tensorflow/python/estimator/BUILD | 121 |
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", + ], +) |