Searched refs:start_indices_ (Results 1 – 1 of 1) sorted by relevance
833 : dim_numbers_(*dim_numbers), start_indices_(*start_indices) { in OutputBatchIndexToInputIndex()852 index_vector_index_.resize(start_indices_.shape().dimensions_size()); in OutputBatchIndexToInputIndex()855 start_indices_.shape().dimensions(dim_numbers_.index_vector_dim()); in OutputBatchIndexToInputIndex()909 start_indices_.GetIntegralAsS64(index_vector_index_)); in FetchIndexVector()947 const Literal& start_indices_; member in xla::OutputBatchIndexToInputIndex