Searched refs:num_classes_with_background (Results 1 – 1 of 1) sorted by relevance
477 const int num_classes_with_background = num_classes + label_offset; in NonMaxSuppressionMultiClassRegularHelper() local497 *(scores + row * num_classes_with_background + col + label_offset); in NonMaxSuppressionMultiClassRegularHelper()508 (selected_index * num_classes_with_background + col + label_offset); in NonMaxSuppressionMultiClassRegularHelper()540 num_classes_with_background); in NonMaxSuppressionMultiClassRegularHelper()543 anchor_index * num_classes_with_background - label_offset; in NonMaxSuppressionMultiClassRegularHelper()601 const int num_classes_with_background = num_classes + label_offset; in NonMaxSuppressionMultiClassFastHelper() local610 scores + row * num_classes_with_background + label_offset; in NonMaxSuppressionMultiClassFastHelper()624 scores + selected_index * num_classes_with_background + label_offset; in NonMaxSuppressionMultiClassFastHelper()647 const int num_classes_with_background, in DequantizeClassPredictions() argument654 for (int idx = 0; idx < num_boxes * num_classes_with_background; ++idx) { in DequantizeClassPredictions()[all …]