Searched refs:kth_element (Results 1 – 1 of 1) sorted by relevance
49 auto kth_element = temp_indices.begin() + k; in TopK() local50 std::partial_sort(temp_indices.begin(), kth_element, temp_indices.end(), in TopK()64 std::copy(temp_indices.begin(), kth_element, out_indices_batch); in TopK()