Searched refs:label_selection_size_ (Results 1 – 2 of 2) 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()
123 label_selection_size_ = label_selection_size; in SetLabelSelectionParameters()147 int label_selection_size_ = 0; // zero means unlimited variable252 (label_selection_size_ > 0 && label_selection_size_ < raw_input.size()); in Step()254 const int max_classes = top_k ? label_selection_size_ : (num_classes_ - 1); in Step()258 max_coeff = GetTopK(label_selection_size_, raw_input, &top_k_logits, in Step()