diff options
-rw-r--r-- | tensorflow/core/kernels/iterator_ops.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/iterator_ops.cc b/tensorflow/core/kernels/iterator_ops.cc index d8bcd09842..9e9d16bbeb 100644 --- a/tensorflow/core/kernels/iterator_ops.cc +++ b/tensorflow/core/kernels/iterator_ops.cc @@ -584,7 +584,7 @@ class OneShotIteratorOp : public AsyncOpKernel { return Status::OK(); } - void ProduceOutput(OpKernelContext* ctx, DoneCallback done) { + void ProduceOutput(OpKernelContext* ctx, const DoneCallback& done) { Tensor* handle; OP_REQUIRES_OK_ASYNC(ctx, ctx->allocate_output(0, TensorShape({}), &handle), done); |