Searched refs:num_categories_per_anchor (Results 1 – 1 of 1) sorted by relevance
602 const int num_categories_per_anchor = in NonMaxSuppressionMultiClassFastHelper() local612 DecreasingPartialArgSort(box_scores, num_classes, num_categories_per_anchor, in NonMaxSuppressionMultiClassFastHelper()628 for (int col = 0; col < num_categories_per_anchor; ++col) { in NonMaxSuppressionMultiClassFastHelper()629 int box_offset = num_categories_per_anchor * output_box_index + col; in NonMaxSuppressionMultiClassFastHelper()