diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-07-14 16:13:52 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-14 16:17:46 -0700 |
| commit | a55f2481ac4e657b73acd606dbc7f85545ef3b2f (patch) | |
| tree | 059c85e1db1f380d957c22e121427571bb4cdb59 /tensorflow/core/kernels/record_yielder.cc | |
| parent | 6254eaeb56d02dc520fc4c055da02e16386adc00 (diff) | |
Fix race in record_yielder.
PiperOrigin-RevId: 162022981
Diffstat (limited to 'tensorflow/core/kernels/record_yielder.cc')
| -rw-r--r-- | tensorflow/core/kernels/record_yielder.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/record_yielder.cc b/tensorflow/core/kernels/record_yielder.cc index 1370295144..e4fa0ed322 100644 --- a/tensorflow/core/kernels/record_yielder.cc +++ b/tensorflow/core/kernels/record_yielder.cc @@ -146,6 +146,7 @@ void RecordYielder::MainLoop() { } if (num_records_added_in_epoch_ < opts_.bufsize) { + mutex_lock l(mu_); opts_.bufsize = num_records_added_in_epoch_; } |