Searched refs:ParallelForWithWorkerId (Results 1 – 6 of 6) sorted by relevance
196 void ParallelForWithWorkerId(202 void ParallelForWithWorkerId(
212 void ThreadPool::ParallelForWithWorkerId( in ParallelForWithWorkerId() function in tensorflow::thread::ThreadPool231 void ThreadPool::ParallelForWithWorkerId( in ParallelForWithWorkerId() function in tensorflow::thread::ThreadPool
218 threads->ParallelForWithWorkerId( in RunFixedBlockSizeShardingWithWorkerId()317 TEST(ThreadPool, ParallelForWithWorkerId) { in TEST() argument332 pool.ParallelForWithWorkerId( in TEST()
64 thread_pool->ParallelForWithWorkerId( in Compute()110 thread_pool->ParallelForWithWorkerId( in Compute()144 thread_pool->ParallelForWithWorkerId( in Compute()
369 worker_threads.workers->ParallelForWithWorkerId( in SparseDenseMatMulWithTransposedLHS()
4192 * Add a `ParallelForWithWorkerId` function.