Home
last modified time | relevance | path

Searched refs:ScopedPerThreadMaxParallelism (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Ddevice_base_test.cc44 ScopedPerThreadMaxParallelism maxp(4); in TEST()
50 ScopedPerThreadMaxParallelism maxp(1); in TEST()
56 ScopedPerThreadMaxParallelism maxp(1000); in TEST()
/external/tensorflow/tensorflow/core/util/
Dwork_sharder.h72 class ScopedPerThreadMaxParallelism {
74 ScopedPerThreadMaxParallelism(int max_parallelism) in ScopedPerThreadMaxParallelism() function
79 ~ScopedPerThreadMaxParallelism() { SetPerThreadMaxParallelism(previous_); } in ~ScopedPerThreadMaxParallelism()
Dwork_sharder_test.cc70 ScopedPerThreadMaxParallelism s(maxp); in TEST()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.cc275 ScopedPerThreadMaxParallelism scope(max_parallelism); in RunnerWithMaxParallelism()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc45 ScopedPerThreadMaxParallelism scope(max_intra_op_parallelism_); in Schedule()