Home
last modified time | relevance | path

Searched refs:num_shards_used (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/platform/
Dthreadpool.cc169 const int num_shards_used = in ParallelForFixedBlockSizeScheduling() local
171 if (num_shards_used == 1) { in ParallelForFixedBlockSizeScheduling()
177 BlockingCounter counter(num_shards_used); in ParallelForFixedBlockSizeScheduling()
191 if (num_shards_used <= NumThreads()) { in ParallelForFixedBlockSizeScheduling()
/external/tensorflow/tensorflow/core/util/
Dwork_sharder.cc83 const int num_shards_used = (total + block_size - 1) / block_size; in Do() local
84 BlockingCounter counter(num_shards_used - 1); in Do()