diff options
author | Derek Murray <mrry@google.com> | 2018-10-08 17:37:44 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-08 17:44:23 -0700 |
commit | 4ff7b81514ea1b86295bc74b620e3c1d3e127e6f (patch) | |
tree | bfa49352d5ff057222784de6cba050548c6c3a03 /tensorflow/python/data/kernel_tests/BUILD | |
parent | 03d097bc96080981098ffdbaf1b3465e6e153a6a (diff) |
Fix the seeding for `Dataset.shuffle(..., reshuffle_each_iteration=False)`.
Previously, we were passing the first (graph-level) seed for both the
graph-level and op-level seeds when creating a C++ dataset. This
change passes the op-level seed to the appropriate point, and adds a test
for the behavior with graph-but-not-op-level seeds.
PiperOrigin-RevId: 216280641
Diffstat (limited to 'tensorflow/python/data/kernel_tests/BUILD')
-rw-r--r-- | tensorflow/python/data/kernel_tests/BUILD | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/python/data/kernel_tests/BUILD b/tensorflow/python/data/kernel_tests/BUILD index c7295d6e69..671b7ca1bb 100644 --- a/tensorflow/python/data/kernel_tests/BUILD +++ b/tensorflow/python/data/kernel_tests/BUILD @@ -443,12 +443,15 @@ tf_py_test( srcs = ["shuffle_dataset_op_test.py"], additional_deps = [ ":test_base", + "@absl_py//absl/testing:parameterized", "//third_party/py/numpy", "//tensorflow/python:array_ops", "//tensorflow/python:client_testlib", "//tensorflow/python:constant_op", "//tensorflow/python:dtypes", "//tensorflow/python:errors", + "//tensorflow/python:framework_ops", + "//tensorflow/python:random_seed", "//tensorflow/python/data/ops:dataset_ops", "//tensorflow/python/data/ops:iterator_ops", ], |