HomeSort by relevance Sort by last modified time
    Searched refs:ParallelTaskAssigner (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
parallel_task_assignment.h 52 // ParallelTaskAssigner computes target parallel task counts for all HLOs
60 class ParallelTaskAssigner : public HloPassInterface {
65 ParallelTaskAssigner(const int64 max_parallelism,
68 ~ParallelTaskAssigner() override {}
parallel_task_assignment.cc 153 StatusOr<bool> ParallelTaskAssigner::Run(HloModule* module) {
154 XLA_VLOG_LINES(2, "ParallelTaskAssigner ENTRY");
164 XLA_VLOG_LINES(2, "ParallelTaskAssigner EXIT");
169 bool ParallelTaskAssigner::AssignParallelTasks(
175 bool ParallelTaskAssigner::AssignParallelTasksHelper(
230 void ParallelTaskAssigner::ComputeTargetParallelTasks(
cpu_compiler.cc 313 // Run ParallelTaskAssigner to assign parallel tasks to HLOs in module.
318 pipeline.AddPass<ParallelTaskAssigner>(max_parallelism,
    [all...]

Completed in 1333 milliseconds