diff options
author | Eugene Brevdo <ebrevdo@google.com> | 2016-08-16 14:20:09 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-08-16 15:33:57 -0700 |
commit | 955efc90026ee08205ad84551dd4a308439fd741 (patch) | |
tree | f31f67c396c99d3b83493a6d833906d67bb47149 /tensorflow/contrib/training/BUILD | |
parent | e7f1beb3468096de0f189fe3be6345a1ef88fe1f (diff) |
Add SequenceQueueingStateSaver to perform truncated reading of sequence minibatches.
Change: 130453767
Diffstat (limited to 'tensorflow/contrib/training/BUILD')
-rw-r--r-- | tensorflow/contrib/training/BUILD | 54 |
1 files changed, 54 insertions, 0 deletions
diff --git a/tensorflow/contrib/training/BUILD b/tensorflow/contrib/training/BUILD new file mode 100644 index 0000000000..5b7046919e --- /dev/null +++ b/tensorflow/contrib/training/BUILD @@ -0,0 +1,54 @@ +# Description: +# contains parts of TensorFlow that are experimental or unstable and which are not supported. + +licenses(["notice"]) # Apache 2.0 + +exports_files(["LICENSE"]) + +package(default_visibility = ["//tensorflow:__subpackages__"]) + +py_library( + name = "training_py", + srcs = [ + "__init__.py", + "python/training/sequence_queueing_state_saver.py", + ], + srcs_version = "PY2AND3", + visibility = ["//visibility:public"], +) + +py_test( + name = "sequence_queueing_state_saver_test", + size = "medium", + srcs = ["python/training/sequence_queueing_state_saver_test.py"], + srcs_version = "PY2AND3", + deps = [ + ":training_py", + "//tensorflow:tensorflow_py", + "//tensorflow/python:framework_test_lib", + ], +) + +py_test( + name = "batch_sequences_with_states_test", + size = "medium", + srcs = ["python/training/batch_sequences_with_states_test.py"], + srcs_version = "PY2AND3", + deps = [ + ":training_py", + "//tensorflow:tensorflow_py", + "//tensorflow/python:framework_test_lib", + ], +) + +filegroup( + name = "all_files", + srcs = glob( + ["**/*"], + exclude = [ + "**/METADATA", + "**/OWNERS", + ], + ), + visibility = ["//tensorflow:__subpackages__"], +) |