Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dmkl_threadpool.h109 class MklDnnThreadPoolWrapper {
111 static MklDnnThreadPoolWrapper& GetInstance() { in GetInstance()
112 static MklDnnThreadPoolWrapper instance_; in GetInstance()
131 MklDnnThreadPoolWrapper() {} in MklDnnThreadPoolWrapper() function
132 MklDnnThreadPoolWrapper(const MklDnnThreadPoolWrapper&) = delete;
133 MklDnnThreadPoolWrapper& operator=(const MklDnnThreadPoolWrapper&) = delete;
134 ~MklDnnThreadPoolWrapper() { in ~MklDnnThreadPoolWrapper()
Dmkl_util.h232 MklDnnThreadPoolWrapper::GetInstance().CreateThreadPoolPtr(ctx); in CreateStream()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_op.cc160 MklDnnThreadPoolWrapper::GetInstance().CreateThreadPoolPtr(ctx); in MklBlasGemm()