Commit d1d58560 authored by Saurabh Saxena's avatar Saurabh Saxena Committed by Gunhan Gulsoy
Browse files

Handle case when buffer_size is larger than total dataset size in shuffle dataset op. 

Currently ShuffleDatasetOpBase::GetNext throws a "Attempted to repeat an empty dataset infinitely." if the number of outputs produced from the dataset are < buffer_size.
Usually the buffer size is set to <= total dataset size so we never run into this.

PiperOrigin-RevId: 182448427
parent dea3facb
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