Searched refs:label_selection_size_ (Results 1 – 1 of 1) sorted by relevance
121 label_selection_size_ = label_selection_size; in SetLabelSelectionParameters()145 int label_selection_size_ = 0; // zero means unlimited variable249 (label_selection_size_ > 0 && label_selection_size_ < raw_input.size()); in Step()251 const int max_classes = top_k ? label_selection_size_ : (num_classes_ - 1); in Step()255 max_coeff = GetTopK(label_selection_size_, raw_input, &top_k_logits, in Step()
Completed in 2 milliseconds