Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dwork_sharder.cc23 /* ABSL_CONST_INIT */ thread_local int per_thread_max_parallelism = 1000000; variable
27 per_thread_max_parallelism = max_parallelism; in SetPerThreadMaxParallelism()
30 int GetPerThreadMaxParallelism() { return per_thread_max_parallelism; } in GetPerThreadMaxParallelism()
Dwork_sharder_test.cc31 int64 per_thread_max_parallelism, in RunSharding() argument
53 if (std::min(num_workers, per_thread_max_parallelism) < in RunSharding()
60 EXPECT_LE(num_shards, 1 + per_thread_max_parallelism); in RunSharding()