Searched refs:index_vector_index_i (Results 1 – 2 of 2) sorted by relevance
1557 int64 index_vector_index_i = 0; in PropagateOutputIndexGatherDimsToIndexVectorIndex() local1563 if (index_vector_index_i == dim_numbers_.index_vector_dim()) { in PropagateOutputIndexGatherDimsToIndexVectorIndex()1564 index_vector_index_i++; in PropagateOutputIndexGatherDimsToIndexVectorIndex()1567 index_vector_index_[index_vector_index_i++] = output_index[i]; in PropagateOutputIndexGatherDimsToIndexVectorIndex()
2185 int64 index_vector_index_i = 0;2191 if (index_vector_index_i == dim_numbers_.index_vector_dim()) {2192 index_vector_index_i++;2195 index_vector_index_[index_vector_index_i++] = update_index[i];