Commit d0e3e998 authored by Saurabh Saxena's avatar Saurabh Saxena Committed by TensorFlower Gardener
Browse files

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
parent 814ab7e3
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment