Searched refs:dimension_partition_counts_ (Results 1 – 2 of 2) sorted by relevance
102 dimension_partition_counts_(dimension_partition_counts), in ShapePartitionIterator()103 dimensions_(dimension_partition_counts_.size()), in ShapePartitionIterator()104 dimension_partition_sizes_(dimension_partition_counts_.size()), in ShapePartitionIterator()105 dimension_partition_strides_(dimension_partition_counts_.size()) { in ShapePartitionIterator()118 std::max(int64{1}, dim_size / dimension_partition_counts_[i]); in ShapePartitionIterator()125 dimension_partition_counts_[i + 1]; in ShapePartitionIterator()141 if (partition_index == dimension_partition_counts_[i] - 1) { in GetPartition()157 dimension_partition_counts_); in GetTotalPartitionCount()
95 const std::vector<int64> dimension_partition_counts_; variable