Searched refs:partition_index (Results 1 – 2 of 2) sorted by relevance
136 const int64 partition_index = index / dimension_partition_strides_[i]; in GetPartition() local138 partition[i].first = partition_index * dimension_partition_sizes_[i]; in GetPartition()141 if (partition_index == dimension_partition_counts_[i] - 1) { in GetPartition()150 index -= partition_index * dimension_partition_strides_[i]; in GetPartition()
293 const int32 partition_index = i * array_partition_stride; in EmitCallToParallelForkJoin() local296 const int32 index = partition_index + j * dim_partition_size; in EmitCallToParallelForkJoin()