Searched defs:topk_indices (Results 1 – 2 of 2) sorted by relevance
124 void TopkAccuracyEvalStage::UpdateCounts(const std::vector<int>& topk_indices) { in UpdateCounts()
164 std::vector<xla::XlaOp> topk_indices; in CreateTopKUnique() local