Searched refs:inter_op_threadpool (Results 1 – 5 of 5) sorted by relevance
148 auto inter_op_threadpool = in TEST() local150 ASSERT_EQ(inter_op_threadpool->GetNumScheduleCalled(), 0); in TEST()157 threadPoolOptions.inter_op_threadpool = inter_op_threadpool.get(); in TEST()216 auto inter_op_threadpool = in TEST() local218 ASSERT_EQ(inter_op_threadpool->GetNumScheduleCalled(), 0); in TEST()225 threadPoolOptions.inter_op_threadpool = inter_op_threadpool.get(); in TEST()252 ASSERT_GT(inter_op_threadpool->GetNumScheduleCalled(), 0); in TEST()
39 tensorflow::thread::ThreadPoolInterface* inter_op_threadpool) in TfThreadPoolWorkQueue() argument41 inter_op_threadpool) {} in TfThreadPoolWorkQueue()45 tensorflow::thread::ThreadPoolInterface* inter_op_threadpool) in TfThreadPoolWorkQueue() argument48 inter_op_threadpool_(inter_op_threadpool) {} in TfThreadPoolWorkQueue()
92 TfThreadPool inter_op_threadpool; in CreateDefaultTfThreadPoolWorkQueue() member96 : inter_op_threadpool("default_work_queue_inter", num_inter_op_threads), in CreateDefaultTfThreadPoolWorkQueue()104 : TfThreadPoolWorkQueue(&thread_pools->inter_op_threadpool, in CreateDefaultTfThreadPoolWorkQueue()
26 thread::ThreadPoolInterface* inter_op_threadpool = nullptr; member
594 } else if (threadpool_options.inter_op_threadpool != nullptr) { in RunInternal()596 threadpool_options.inter_op_threadpool); in RunInternal()