HomeSort by relevance Sort by last modified time
    Searched defs:max_parallelism (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
parallel_task_assignment.cc 30 SimpleCostModel(const int64 max_parallelism,
32 : max_parallelism_(max_parallelism), shape_size_(shape_size) {}
51 DefaultCostModel(const int64 max_parallelism,
54 : max_parallelism_(max_parallelism),
63 int64 max_parallelism; variable
75 max_parallelism =
82 max_parallelism = max_parallelism_;
95 return std::min(max_parallelism,
106 const int64 max_parallelism,
108 VLOG(1) << "ParallelTaskAssignment max_parallelism: " << max_parallelism
    [all...]
cpu_compiler.cc 305 const int max_parallelism = local
310 pipeline.AddPass<ParallelizationPreparation>(max_parallelism,
318 pipeline.AddPass<ParallelTaskAssigner>(max_parallelism,
333 pipeline.AddPass<ParallelizationPreparation>(max_parallelism,
    [all...]

Completed in 223 milliseconds