aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/training/BUILD
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2016-11-02 12:53:35 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-11-02 14:12:15 -0700
commit7c89c44112a50d0fcf8888f14c799eefda841454 (patch)
treed33d27884f52e85c5733e6caa276849f8456c33a /tensorflow/contrib/training/BUILD
parent2f8318865e248eb1699170a5588567e6d9b50b05 (diff)
Update generated Python Op docs.
Change: 137988710
Diffstat (limited to 'tensorflow/contrib/training/BUILD')
-rw-r--r--tensorflow/contrib/training/BUILD20
1 files changed, 20 insertions, 0 deletions
diff --git a/tensorflow/contrib/training/BUILD b/tensorflow/contrib/training/BUILD
index 81dc8e9064..9870c37d19 100644
--- a/tensorflow/contrib/training/BUILD
+++ b/tensorflow/contrib/training/BUILD
@@ -15,9 +15,16 @@ py_library(
"python/training/resample.py",
"python/training/sampling_ops.py",
"python/training/sequence_queueing_state_saver.py",
+ "python/training/training.py",
],
srcs_version = "PY2AND3",
visibility = ["//visibility:public"],
+ deps = [
+ "//tensorflow/python:framework",
+ "//tensorflow/python:ops",
+ "//tensorflow/python:platform",
+ "//tensorflow/python:training",
+ ],
)
py_test(
@@ -93,6 +100,19 @@ py_test(
],
)
+py_test(
+ name = "training_test",
+ size = "large",
+ srcs = ["python/training/training_test.py"],
+ shard_count = 3,
+ srcs_version = "PY2AND3",
+ deps = [
+ ":training_py",
+ "//tensorflow:tensorflow_py",
+ "//tensorflow/python:framework_test_lib",
+ ],
+)
+
filegroup(
name = "all_files",
srcs = glob(