Searched refs:dimension_partition_strides_ (Results 1 – 2 of 2) sorted by relevance
105 dimension_partition_strides_(dimension_partition_counts_.size()) { in ShapePartitionIterator()122 dimension_partition_strides_[dimension_partition_strides_.size() - 1] = 1; in ShapePartitionIterator()123 for (int i = dimension_partition_strides_.size() - 2; i >= 0; --i) { in ShapePartitionIterator()124 dimension_partition_strides_[i] = dimension_partition_strides_[i + 1] * in ShapePartitionIterator()136 const int64 partition_index = index / dimension_partition_strides_[i]; in GetPartition()150 index -= partition_index * dimension_partition_strides_[i]; in GetPartition()
99 std::vector<int64> dimension_partition_strides_; variable