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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
parallel_task_assignment.h 33 // ParallelTaskAssignment computes parallel task counts for HLOs in 'module'.
34 class ParallelTaskAssignment {
40 ParallelTaskAssignment(const int64 max_parallelism,
43 ~ParallelTaskAssignment() {}
parallel_task_assignment.cc 105 ParallelTaskAssignment::ParallelTaskAssignment(
108 VLOG(1) << "ParallelTaskAssignment max_parallelism: " << max_parallelism;
125 int64 ParallelTaskAssignment::GetTargetParallelTaskCount(
232 ParallelTaskAssignment parallel_task_assignment(max_parallelism_,
241 // Query ParallelTaskAssignment for target parallel task count.
cpu_parallelization_preparation.cc 113 // Initialize ParallelTaskAssignment.
114 ParallelTaskAssignment parallel_task_assignment(max_parallelism_, shape_size_,

Completed in 65 milliseconds