Fix msan error in MapAndBatchDataset.
authorSaurabh Saxena <srbs@google.com>
Fri, 20 Apr 2018 16:23:52 +0000 (09:23 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 20 Apr 2018 16:26:42 +0000 (09:26 -0700)
commitd0e3e998376f5e7d59678e5d42f3497e52ca7622
tree4966b2828ba4ce590fe168d73cafcc5b0bbc7d0c
parent814ab7e37dcbfa7f4749a1fd9d687d6be0207cb8
Fix msan error in MapAndBatchDataset.
While checkpointing tensors in BatchResult.output save only the initialized slice.
If the final batch is short, the entire batch tensor may not be initialized.

PiperOrigin-RevId: 193678679
tensorflow/core/kernels/data/map_and_batch_dataset_op.cc