Searched refs:index_vector_index_ (Results 1 – 2 of 2) sorted by relevance
1521 index_vector_index_.resize(start_indices_.shape().dimensions_size()); in OutputBatchIndexToInputIndex()1567 index_vector_index_[index_vector_index_i++] = output_index[i]; in PropagateOutputIndexGatherDimsToIndexVectorIndex()1576 index_vector_index_[index_vector_dim] = i; in FetchIndexVector()1577 auto start_index = start_indices_.GetIntegralAsS64(index_vector_index_); in FetchIndexVector()1607 std::vector<int64> index_vector_index_; member in xla::OutputBatchIndexToInputIndex
2149 index_vector_index_.resize(scatter_indices_.shape().dimensions_size());2195 index_vector_index_[index_vector_index_i++] = update_index[i];2204 index_vector_index_[index_vector_dim] = i;2206 *scatter_indices_.GetIntegralAsS64(index_vector_index_);2234 std::vector<int64> index_vector_index_;