Home
last modified time | relevance | path

Searched refs:ParallelTaskAssigner (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dparallel_task_assignment.h64 class ParallelTaskAssigner : public HloModulePass {
69 ParallelTaskAssigner(const int64_t max_parallelism, in ParallelTaskAssigner() function
75 ~ParallelTaskAssigner() override {} in ~ParallelTaskAssigner()
H A Dparallel_task_assignment.cc175 StatusOr<bool> ParallelTaskAssigner::Run( in Run()
193 bool ParallelTaskAssigner::AssignParallelTasks( in AssignParallelTasks()
199 bool ParallelTaskAssigner::AssignParallelTasksHelper( in AssignParallelTasksHelper()
253 void ParallelTaskAssigner::ComputeTargetParallelTasks( in ComputeTargetParallelTasks()
H A Dparallel_task_assignment_test.cc44 return cpu::ParallelTaskAssigner(max_parallelism_, shape_size_func_, in RunParallelTaskAssigner()
H A Dcpu_compiler.cc682 pipeline.AddPass<ParallelTaskAssigner>( in RunHloPassesAfterLayoutAssn()