diff --git a/tensorflow/core/kernels/record_input_op.cc b/tensorflow/core/kernels/record_input_op.cc index 60c0a7d2d8c..878996c9d6a 100644 --- a/tensorflow/core/kernels/record_input_op.cc +++ b/tensorflow/core/kernels/record_input_op.cc @@ -53,7 +53,7 @@ class RecordInputOp : public OpKernel { Tensor out(DT_STRING, {batch_size_}); auto t_out = out.flat(); for (int i = 0; i < batch_size_; ++i) { - OP_REQUIRES_OK(ctx, yielder_->Yield(&t_out(i))); + OP_REQUIRES_OK(ctx, yielder_->YieldOne(&t_out(i))); } ctx->set_output(0, out); } diff --git a/tensorflow/core/kernels/record_yielder.cc b/tensorflow/core/kernels/record_yielder.cc index d65ba2c0260..e391752289b 100644 --- a/tensorflow/core/kernels/record_yielder.cc +++ b/tensorflow/core/kernels/record_yielder.cc @@ -43,7 +43,7 @@ RecordYielder::~RecordYielder() { delete thread_; } -Status RecordYielder::Yield(string* value) { +Status RecordYielder::YieldOne(string* value) { mutex_lock l(mu_); while (!BufEnough()) { buf_enough_.wait(l); diff --git a/tensorflow/core/kernels/record_yielder.h b/tensorflow/core/kernels/record_yielder.h index d331995e47f..503644f3b87 100644 --- a/tensorflow/core/kernels/record_yielder.h +++ b/tensorflow/core/kernels/record_yielder.h @@ -50,7 +50,7 @@ namespace tensorflow { // RecordYielder yielder(opts); // string val; // while (true) { -// yielder.Yield(&val); +// yielder.YieldOne(&val); // // process val // } // @@ -88,7 +88,7 @@ class RecordYielder { RecordYielder& operator=(const RecordYielder&) = delete; // Yields one 'value'. - Status Yield(string* value); + Status YieldOne(string* value); // Returns the current epoch number. int64 current_epoch() const { return epoch_; }