avoid unnecessary copy (repeat 8834e4f)
diff --git a/tensorflow/core/kernels/data/dataset_test_base.cc b/tensorflow/core/kernels/data/dataset_test_base.cc
index ce194a8..6b3df82 100644
--- a/tensorflow/core/kernels/data/dataset_test_base.cc
+++ b/tensorflow/core/kernels/data/dataset_test_base.cc
@@ -333,7 +333,7 @@
nullptr /* cluster_flr */);
flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0");
if (thread_pool_ == nullptr) {
- runner_ = [](std::function<void()> fn) { fn(); };
+ runner_ = [](const std::function<void()> fn) { fn(); };
} else {
runner_ = [this](std::function<void()> fn) {
thread_pool_->Schedule(std::move(fn));
diff --git a/tensorflow/core/kernels/data/single_threaded_executor_test.cc b/tensorflow/core/kernels/data/single_threaded_executor_test.cc
index 19a3a65..84838a1 100644
--- a/tensorflow/core/kernels/data/single_threaded_executor_test.cc
+++ b/tensorflow/core/kernels/data/single_threaded_executor_test.cc
@@ -68,7 +68,7 @@
};
delete exec_;
TF_CHECK_OK(NewSingleThreadedExecutor(params, *graph, &exec_));
- runner_ = [](std::function<void()> fn) { fn(); };
+ runner_ = [](const std::function<void()> fn) { fn(); };
rendez_ = NewLocalRendezvous();
}
