Home
last modified time | relevance | path

Searched refs:threadpool_map_ (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dmkl_threadpool.h117 if (threadpool_map_.empty() || in CreateThreadPoolPtr()
118 threadpool_map_.find(ctx->device()) == threadpool_map_.end()) { in CreateThreadPoolPtr()
120 threadpool_map_.emplace(std::make_pair(ctx->device(), tp_iface)); in CreateThreadPoolPtr()
123 auto entry = threadpool_map_.find(ctx->device()); in CreateThreadPoolPtr()
130 std::unordered_map<DeviceBase*, MklDnnThreadPool*> threadpool_map_; variable
135 for (auto& tp : threadpool_map_) { in ~MklDnnThreadPoolWrapper()