Home
last modified time | relevance | path

Searched refs:total_partition_count (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dshape_partition.cc92 int64 total_partition_count = 1; in GetTotalPartitionCount() local
94 total_partition_count *= dim_partition_count; in GetTotalPartitionCount()
96 return total_partition_count; in GetTotalPartitionCount()
Dparallel_task_assignment.cc211 const int64 total_partition_count = in AssignParallelTasksHelper() local
213 if (total_partition_count <= 1) { in AssignParallelTasksHelper()
227 VLOG(2) << "Assigned parallel task count: " << total_partition_count in AssignParallelTasksHelper()