Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dshape_partition.cc102 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()
Dshape_partition.h95 const std::vector<int64> dimension_partition_counts_; variable