Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dwork_sharder.h74 class ScopedPerThreadMaxParallelism {
76 ScopedPerThreadMaxParallelism(int max_parallelism) in ScopedPerThreadMaxParallelism() function
81 ~ScopedPerThreadMaxParallelism() { SetPerThreadMaxParallelism(previous_); } in ~ScopedPerThreadMaxParallelism()
Dwork_sharder_test.cc70 ScopedPerThreadMaxParallelism s(maxp); in TEST()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base_test.cc43 ScopedPerThreadMaxParallelism maxp(4); in TEST()
49 ScopedPerThreadMaxParallelism maxp(1); in TEST()
55 ScopedPerThreadMaxParallelism maxp(1000); in TEST()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc48 ScopedPerThreadMaxParallelism scope(max_intra_op_parallelism_); in Schedule()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.cc476 ScopedPerThreadMaxParallelism scope(max_parallelism); in RunnerWithMaxParallelism()