Home
last modified time | relevance | path

Searched refs:GetPerThreadMaxParallelism (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dwork_sharder.h71 int GetPerThreadMaxParallelism();
77 : previous_(GetPerThreadMaxParallelism()) { in ScopedPerThreadMaxParallelism()
Dwork_sharder.cc30 int GetPerThreadMaxParallelism() { return per_thread_max_parallelism; } in GetPerThreadMaxParallelism() function
38 max_parallelism = std::min(max_parallelism, GetPerThreadMaxParallelism()); in Shard()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.cc97 std::min<int>(GetPerThreadMaxParallelism(), eigen_cpu_devices_.size())); in eigen_cpu_device()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils_test.cc269 auto fn = []() { ASSERT_EQ(GetPerThreadMaxParallelism(), 2); }; in TEST()