diff options
| author |  Rohan Jain <rohanj@google.com> | 2018-09-28 08:26:55 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-28 08:38:03 -0700 |
| commit | 4e955be2ae1c920623778c15357129fea9a3bdab (patch) | |
| tree | 1c5712af9c3fb4b71ae23a9dc22c3415f953d9b7 /models.BUILD | |
| parent | 32627bfba19606d3c3a34f5d02ae9428675bbc42 (diff) | |
Fixing a couple of small bugs with the multi device iterator having to deal
with the case when the background thread terminated because the iterator
finished and yet some other requests were coming in.
1. The GetNextFromShard would see an empty buffer and return cancelled instead
of OutOfRange errors
2. On shutdown, we weren't calling all the pending callbacks.
Tested with runs_per_test=5000
PiperOrigin-RevId: 214939274
Diffstat (limited to 'models.BUILD')
0 files changed, 0 insertions, 0 deletions