diff options
Diffstat (limited to 'tensorflow/python/estimator/BUILD')
| -rw-r--r-- | tensorflow/python/estimator/BUILD | 75 |
1 files changed, 64 insertions, 11 deletions
diff --git a/tensorflow/python/estimator/BUILD b/tensorflow/python/estimator/BUILD index 2d9a084bc6..d7a9dc5ede 100644 --- a/tensorflow/python/estimator/BUILD +++ b/tensorflow/python/estimator/BUILD @@ -21,6 +21,7 @@ py_library( ":export", ":exporter", ":inputs", + ":keras", ":linear", ":model_fn", ":parsing_utils", @@ -445,16 +446,6 @@ py_library( ], ) -py_test( - name = "util_test", - srcs = ["util_test.py"], - srcs_version = "PY2AND3", - deps = [ - ":util", - "//tensorflow/python:client_testlib", - ], -) - py_library( name = "estimator", srcs = [ @@ -645,7 +636,6 @@ py_library( ":metric_keys", ":model_fn", ":prediction_keys", - ":util", "//tensorflow/python:array_ops", "//tensorflow/python:check_ops", "//tensorflow/python:control_flow_ops", @@ -659,6 +649,7 @@ py_library( "//tensorflow/python:string_ops", "//tensorflow/python:summary", "//tensorflow/python:training", + "//tensorflow/python:util", "//tensorflow/python:weights_broadcast_ops", "//tensorflow/python/feature_column", "//tensorflow/python/ops/losses", @@ -911,3 +902,65 @@ py_test( "//tensorflow/python:training", ], ) + +py_library( + name = "keras", + srcs = ["keras.py"], + srcs_version = "PY2AND3", + deps = [ + ":estimator", + ":export_export", + ":model_fn", + ":run_config", + "//tensorflow/python:check_ops", + "//tensorflow/python:framework_ops", + "//tensorflow/python:init_ops", + "//tensorflow/python:layers", + "//tensorflow/python:math_ops", + "//tensorflow/python:metrics", + "//tensorflow/python:nn", + "//tensorflow/python:partitioned_variables", + "//tensorflow/python:platform", + "//tensorflow/python:random_seed", + "//tensorflow/python:session", + "//tensorflow/python:sparse_tensor", + "//tensorflow/python:summary", + "//tensorflow/python:tensor_util", + "//tensorflow/python:training", + "//tensorflow/python:training_util", + "//tensorflow/python:util", + "//tensorflow/python:variable_scope", + "//tensorflow/python:variables", + "//tensorflow/python/feature_column", + "//tensorflow/python/keras:backend", + "//tensorflow/python/keras:engine", + "//tensorflow/python/keras:layers", + "//tensorflow/python/ops/losses", + "//tensorflow/python/saved_model", + "//tensorflow/python/saved_model:signature_constants", + ], +) + +py_test( + name = "keras_test", + size = "large", + srcs = ["keras_test.py"], + srcs_version = "PY2AND3", + tags = ["notsan"], + deps = [ + ":keras", + "//tensorflow/core:protos_all_py", + "//tensorflow/python:client_testlib", + "//tensorflow/python:framework_ops", + "//tensorflow/python:framework_test_lib", + "//tensorflow/python:platform", + "//tensorflow/python:summary", + "//tensorflow/python:training", + "//tensorflow/python/estimator:numpy_io", + "//tensorflow/python/estimator:run_config", + "//tensorflow/python/keras", + "//tensorflow/python/keras:backend", + "//tensorflow/python/keras:engine", + "//third_party/py/numpy", + ], +) |