Searched refs:ParallelForWithWorkerId (Results 1 – 6 of 6) sorted by relevance
197 void ParallelForWithWorkerId(203 void ParallelForWithWorkerId(
212 void ThreadPool::ParallelForWithWorkerId( in ParallelForWithWorkerId() function in tensorflow::thread::ThreadPool231 void ThreadPool::ParallelForWithWorkerId( in ParallelForWithWorkerId() function in tensorflow::thread::ThreadPool
219 threads->ParallelForWithWorkerId( in RunFixedBlockSizeShardingWithWorkerId()318 TEST(ThreadPool, ParallelForWithWorkerId) { in TEST() argument333 pool.ParallelForWithWorkerId( in TEST()
66 thread_pool->ParallelForWithWorkerId( in Compute()139 thread_pool->ParallelForWithWorkerId( in Compute()179 thread_pool->ParallelForWithWorkerId( in Compute()
378 worker_threads.workers->ParallelForWithWorkerId( in SparseDenseMatMulWithTransposedLHS()
9515 * Add a `ParallelForWithWorkerId` function.