Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dshape_partition_test.cc187 const auto& dim_partition = partition_iterator.GetPartition(i); in TEST_F() local
188 for (int dim = 0; dim < dim_partition.size(); ++dim) { in TEST_F()
190 std::make_pair(dim_partition[dim].first, in TEST_F()
191 dim_partition[dim].first + dim_partition[dim].second)); in TEST_F()
Dir_function.cc302 const std::pair<int64, int64>& dim_partition = dim_partitions[j]; in EmitCallToParallelForkJoin() local
305 partitions[index] = b->getInt64(dim_partition.first); in EmitCallToParallelForkJoin()
307 b->getInt64(dim_partition.first + dim_partition.second); in EmitCallToParallelForkJoin()