diff options
author | 2016-11-02 12:53:35 -0800 | |
---|---|---|
committer | 2016-11-02 14:12:15 -0700 | |
commit | 7c89c44112a50d0fcf8888f14c799eefda841454 (patch) | |
tree | d33d27884f52e85c5733e6caa276849f8456c33a /tensorflow/contrib/training/BUILD | |
parent | 2f8318865e248eb1699170a5588567e6d9b50b05 (diff) |
Update generated Python Op docs.
Change: 137988710
Diffstat (limited to 'tensorflow/contrib/training/BUILD')
-rw-r--r-- | tensorflow/contrib/training/BUILD | 20 |
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( |