[tf.data] Allow `sample_from_datasets` to accept a tf.Dataset object for `weights`.
authorjoel-shor <joelshor@google.com>
Thu, 19 Apr 2018 22:41:28 +0000 (01:41 +0300)
committerjoel-shor <joelshor@google.com>
Thu, 19 Apr 2018 22:41:28 +0000 (01:41 +0300)
commitf500bcb889b3598f386f59eb69a79af6b704bf50
treeaa96367727e3b3dbf70e4356f2d8b392bd1e957f
parent4ca24c6e7df49881ae5770c331a29778a43d3c4c
[tf.data] Allow `sample_from_datasets` to accept a tf.Dataset object for `weights`.

Tested:
bazel test :interleave_dataset_op_test
tensorflow/contrib/data/python/kernel_tests/interleave_dataset_op_test.py
tensorflow/contrib/data/python/ops/interleave_ops.py