Searched refs:GetTotalPartitionCount (Results 1 – 5 of 5) sorted by relevance
62 if (GetTotalPartitionCount(dimension_partition_counts) < in Run()70 GetTotalPartitionCount(dimension_partition_counts) / in Run()90 int64 ShapePartitionAssigner::GetTotalPartitionCount( in GetTotalPartitionCount() function in xla::cpu::ShapePartitionAssigner155 int64 ShapePartitionIterator::GetTotalPartitionCount() const { in GetTotalPartitionCount() function in xla::cpu::ShapePartitionIterator156 return ShapePartitionAssigner::GetTotalPartitionCount( in GetTotalPartitionCount()
104 EXPECT_EQ(1, iterator.GetTotalPartitionCount()); in TEST_F()110 EXPECT_EQ(2, iterator.GetTotalPartitionCount()); in TEST_F()117 EXPECT_EQ(3, iterator.GetTotalPartitionCount()); in TEST_F()129 EXPECT_EQ(1, iterator.GetTotalPartitionCount()); in TEST_F()136 EXPECT_EQ(4, iterator.GetTotalPartitionCount()); in TEST_F()185 const int64 partition_count = partition_iterator.GetTotalPartitionCount(); in TEST_F()
73 static int64 GetTotalPartitionCount(91 int64 GetTotalPartitionCount() const;
212 ShapePartitionAssigner::GetTotalPartitionCount(dim_partition_counts); in AssignParallelTasksHelper()
271 const int64 num_partitions = partition_iterator.GetTotalPartitionCount(); in EmitCallToParallelForkJoin()