diff options
author | 2016-10-27 21:53:15 -0800 | |
---|---|---|
committer | 2016-10-27 23:08:56 -0700 | |
commit | 22b062cf921ccce34e0cf912c4c750d5b9ba3d10 (patch) | |
tree | 4c60c53f3a0f41f2803353e1e636beb0dc987cb5 /tensorflow/cc/training | |
parent | e3fab91840ab7709ac3c9b4aa0155bc78bda8ecc (diff) |
Santinize thread pool name in C++ QueueRunner.
Change: 137481049
Diffstat (limited to 'tensorflow/cc/training')
-rw-r--r-- | tensorflow/cc/training/queue_runner.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/cc/training/queue_runner.cc b/tensorflow/cc/training/queue_runner.cc index 79d306f367..216abb7236 100644 --- a/tensorflow/cc/training/queue_runner.cc +++ b/tensorflow/cc/training/queue_runner.cc @@ -14,6 +14,7 @@ limitations under the License. ==============================================================================*/ #include "tensorflow/cc/training/queue_runner.h" +#include "tensorflow/core/kernels/ops_util.h" #include "tensorflow/core/platform/env.h" namespace tensorflow { @@ -47,8 +48,8 @@ Status QueueRunner::Init(const QueueRunnerDef& queue_runner_def) { } } - thread_pool_.reset( - new thread::ThreadPool(Env::Default(), queue_name_, runs_)); + thread_pool_.reset(new thread::ThreadPool( + Env::Default(), SanitizeThreadSuffix(queue_name_), runs_)); should_stop_ = false; return Status::OK(); } |