Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/lib/core/
Dthreadpool.cc145 const int num_shards_used = in TransformRangeConcurrently() local
147 if (num_shards_used == 1) { in TransformRangeConcurrently()
153 BlockingCounter counter(num_shards_used); in TransformRangeConcurrently()
167 if (num_shards_used <= NumThreads()) { in TransformRangeConcurrently()
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dparallel_for.cc34 const int num_shards_used = (batch_size + block_size - 1) / block_size; in ParallelFor() local
35 BlockingCounter counter(num_shards_used - 1); in ParallelFor()
/external/tensorflow/tensorflow/core/util/
Dwork_sharder.cc81 const int num_shards_used = (total + block_size - 1) / block_size; in Do() local
82 BlockingCounter counter(num_shards_used - 1); in Do()