Searched refs:top_k_indices (Results 1 – 3 of 3) sorted by relevance
112 std::vector<int>* top_k_indices);217 std::vector<int>* top_k_indices) { in GetTopK() argument222 top_k_indices->clear(); in GetTopK()224 top_k_indices->resize(K, -1); in GetTopK()231 (*top_k_indices)[k] = (*top_k_indices)[k - 1]; in GetTopK()235 (*top_k_indices)[k] = j; in GetTopK()247 std::vector<int> top_k_indices; in Step() local256 &top_k_indices); in Step()335 const int label = top_k ? top_k_indices[ind] : ind; in Step()
114 std::vector<int>* top_k_indices);220 std::vector<int>* top_k_indices) { in GetTopK() argument225 top_k_indices->clear(); in GetTopK()227 top_k_indices->resize(K, -1); in GetTopK()234 (*top_k_indices)[k] = (*top_k_indices)[k - 1]; in GetTopK()238 (*top_k_indices)[k] = j; in GetTopK()250 std::vector<int> top_k_indices; in Step() local259 &top_k_indices); in Step()338 const int label = top_k ? top_k_indices[ind] : ind; in Step()
274 int* top_k_indices) {324 top_k_indices[rank] = entries[shard_index].index;336 top_k_indices[last_k] = entries[shard_index].index;