Searched refs:dimension_partition_sizes_ (Results 1 – 2 of 2) sorted by relevance
104 dimension_partition_sizes_(dimension_partition_counts_.size()), in ShapePartitionIterator()115 for (int i = 0; i < dimension_partition_sizes_.size(); ++i) { in ShapePartitionIterator()117 dimension_partition_sizes_[i] = in ShapePartitionIterator()138 partition[i].first = partition_index * dimension_partition_sizes_[i]; in GetPartition()146 partition[i].second = dimension_partition_sizes_[i]; in GetPartition()
98 std::vector<int64> dimension_partition_sizes_; variable