Home
last modified time | relevance | path

Searched defs:max_parallelism (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dwork_sharder.cc25 void SetPerThreadMaxParallelism(int max_parallelism) { in SetPerThreadMaxParallelism()
32 void Shard(int max_parallelism, thread::ThreadPool* workers, int64 total, in Shard()
56 const Runner& runner, int max_parallelism) { in Do()
Dwork_sharder.h74 ScopedPerThreadMaxParallelism(int max_parallelism) in ScopedPerThreadMaxParallelism()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_task_assignment.cc32 SimpleCostModel(const int64 max_parallelism, in SimpleCostModel()
53 DefaultCostModel(const int64 max_parallelism, in DefaultCostModel()
65 int64 max_parallelism; in GetParallelTaskCount() local
108 const int64 max_parallelism, in ParallelTaskAssignment()
Dparallel_task_assignment.h68 ParallelTaskAssigner(const int64 max_parallelism, in ParallelTaskAssigner()
Dcpu_compiler.cc368 const int max_parallelism = in RunHloPassesAfterLayoutAssn() local
/external/tensorflow/tensorflow/core/kernels/
Dredux_functor.h50 const int64 max_parallelism = total_workload / parallel_cell_size; in operator() local
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.cc267 std::function<void(std::function<void()>)> runner, int max_parallelism) { in RunnerWithMaxParallelism()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc310 auto max_parallelism = dataset()->max_intra_op_parallelism_; in GetNextInternal() local