Searched refs:parallel_idx (Results 1 – 1 of 1) sorted by relevance
908 for (int i = 0, parallel_idx = 0; i < gather_shape.rank(); ++i) { in GatherParallelDataOperandSharding() local909 if (parallel_idx >= output_parallel_dims.size() || in GatherParallelDataOperandSharding()910 output_parallel_dims[parallel_idx] != i) { in GatherParallelDataOperandSharding()914 output_aligned_operand_parallel_dims[parallel_idx++]; in GatherParallelDataOperandSharding()