Added `drop_final_batch` argument to make_batched_features_dataset. This allows the...
authorA. Unique TensorFlower <gardener@tensorflow.org>
Fri, 6 Apr 2018 01:37:19 +0000 (18:37 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 6 Apr 2018 01:39:37 +0000 (18:39 -0700)
commit96a72a9a836ad1c1f46e74d30eda4deb78740efe
tree7b133d8483f16c57d196364e64786669a3443c5d
parent2248a3488c53f8b858e2a0b8be93d62c3056df36
Added `drop_final_batch` argument to make_batched_features_dataset. This allows the batch_and_drop_remainder function to be used instead of the default batch function.

PiperOrigin-RevId: 191831842
tensorflow/contrib/data/python/kernel_tests/reader_dataset_ops_test.py
tensorflow/contrib/data/python/ops/readers.py