Searched defs:MklDnnThreadPool (Results 1 – 1 of 1) sorted by relevance
61 struct MklDnnThreadPool : public dnnl::threadpool_iface { struct64 MklDnnThreadPool(OpKernelContext* ctx) in MklDnnThreadPool() function68 virtual int get_num_threads() const override { in get_num_threads()71 virtual bool get_in_parallel() const override { in get_in_parallel()74 virtual uint64_t get_flags() const override { return ASYNCHRONOUS; } in get_flags()75 virtual void parallel_for(int n, in parallel_for()103 ~MklDnnThreadPool() {} in ~MklDnnThreadPool() argument