diff options
author | 2016-09-28 09:13:02 -0800 | |
---|---|---|
committer | 2016-09-28 10:19:02 -0700 | |
commit | 09ecbd94c745361142d1d2a3cab39d66892c9013 (patch) | |
tree | 67985d6206afcb4a8c0f43a6ff9bad265dd3f239 /tensorflow/contrib/training/BUILD | |
parent | f4c795b7e60f563ae21baffec1870f34178b914a (diff) |
Adds a method to resample tensors from of batches at either a per-item specific rate, or at a rate proportional to per-item weights.
Change: 134548493
Diffstat (limited to 'tensorflow/contrib/training/BUILD')
-rw-r--r-- | tensorflow/contrib/training/BUILD | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/contrib/training/BUILD b/tensorflow/contrib/training/BUILD index 79901b6ee5..81dc8e9064 100644 --- a/tensorflow/contrib/training/BUILD +++ b/tensorflow/contrib/training/BUILD @@ -12,6 +12,7 @@ py_library( srcs = [ "__init__.py", "python/training/bucket_ops.py", + "python/training/resample.py", "python/training/sampling_ops.py", "python/training/sequence_queueing_state_saver.py", ], @@ -44,6 +45,18 @@ py_test( ) py_test( + name = "resample_test", + size = "small", + srcs = ["python/training/resample_test.py"], + srcs_version = "PY2AND3", + deps = [ + ":training_py", + "//tensorflow:tensorflow_py", + "//tensorflow/python:framework_test_lib", + ], +) + +py_test( name = "sampling_ops_test", size = "small", srcs = ["python/training/sampling_ops_test.py"], |