Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dshape_partition.cc93 for (int64 dim_partition_count : dimension_partition_counts) { in GetTotalPartitionCount() local
94 total_partition_count *= dim_partition_count; in GetTotalPartitionCount()
Dshape_partition_test.cc178 const int64 dim_partition_count = 1 + Rand() % dim_sizes[i]; in TEST_F() local
179 dim_partition_counts[i] = dim_partition_count; in TEST_F()