Searched refs:MklDnnThreadPool (Results 1 – 1 of 1) sorted by relevance
61 struct MklDnnThreadPool : public dnnl::threadpool_iface { struct62 MklDnnThreadPool() = default;64 MklDnnThreadPool(OpKernelContext* ctx) in MklDnnThreadPool() function103 ~MklDnnThreadPool() {} in ~MklDnnThreadPool() argument115 MklDnnThreadPool* CreateThreadPoolPtr(OpKernelContext* ctx) { in CreateThreadPoolPtr()119 auto tp_iface = new MklDnnThreadPool(ctx); in CreateThreadPoolPtr()130 std::unordered_map<DeviceBase*, MklDnnThreadPool*> threadpool_map_;