Searched refs:start_indices_ (Results 1 – 1 of 1) sorted by relevance
1502 : dim_numbers_(*dim_numbers), start_indices_(*start_indices) { in OutputBatchIndexToInputIndex()1521 index_vector_index_.resize(start_indices_.shape().dimensions_size()); in OutputBatchIndexToInputIndex()1524 start_indices_.shape().dimensions(dim_numbers_.index_vector_dim()); in OutputBatchIndexToInputIndex()1577 auto start_index = start_indices_.GetIntegralAsS64(index_vector_index_); in FetchIndexVector()1617 const Literal& start_indices_; member in xla::OutputBatchIndexToInputIndex