Searched refs:max_classes (Results 1 – 2 of 2) sorted by relevance
251 const int max_classes = top_k ? label_selection_size_ : (num_classes_ - 1); in Step() local334 for (int ind = 0; ind < max_classes; ind++) { in Step()
254 const int max_classes = top_k ? label_selection_size_ : (num_classes_ - 1); in Step() local337 for (int ind = 0; ind < max_classes; ind++) { in Step()