Searched refs:index_vector_dim_index (Results 1 – 2 of 2) sorted by relevance
132 int64 index_vector_dim_index = FindIndex(dim_numbers.start_index_map(), i); in ExpandIndexVectorIntoOperandSpace() local133 if (index_vector_dim_index != dim_numbers.start_index_map_size()) { in ExpandIndexVectorIntoOperandSpace()136 MakeSliceHlo(index_vector, /*start_indices=*/{index_vector_dim_index}, in ExpandIndexVectorIntoOperandSpace()137 /*limit_indices=*/{index_vector_dim_index + 1}, in ExpandIndexVectorIntoOperandSpace()
153 int64 index_vector_dim_index = in ExpandIndexVectorIntoOperandSpace() local155 if (index_vector_dim_index != in ExpandIndexVectorIntoOperandSpace()159 MakeSliceHlo(index_vector, /*start_indices=*/{index_vector_dim_index}, in ExpandIndexVectorIntoOperandSpace()160 /*limit_indices=*/{index_vector_dim_index + 1}, in ExpandIndexVectorIntoOperandSpace()