Searched refs:index_vector_index_ (Results 1 – 2 of 2) sorted by relevance
852 index_vector_index_.resize(start_indices_.shape().dimensions_size()); in OutputBatchIndexToInputIndex()898 index_vector_index_[index_vector_index_i++] = output_index[i]; in PropagateOutputIndexGatherDimsToIndexVectorIndex()907 index_vector_index_[index_vector_dim] = i; in FetchIndexVector()909 start_indices_.GetIntegralAsS64(index_vector_index_)); in FetchIndexVector()937 std::vector<int64> index_vector_index_; member in xla::OutputBatchIndexToInputIndex
2126 index_vector_index_.resize(scatter_indices_.shape().dimensions_size()); in UpdateScatterIndexToInputIndex()2172 index_vector_index_[index_vector_index_i++] = update_index[i]; in PropagateUpdateIndexScatterDimsToIndexVectorIndex()2181 index_vector_index_[index_vector_dim] = i; in FetchIndexVector()2183 index_vector_index_)); in FetchIndexVector()2211 std::vector<int64> index_vector_index_; variable