Searched refs:sorted_indices (Results 1 – 2 of 2) sorted by relevance
434 int* sorted_indices = reinterpret_cast<int*>( in NonMaxSuppressionSingleClassHelper() local438 sorted_indices); in NonMaxSuppressionSingleClassHelper()454 selected[(*selected_size)++] = keep_indices[sorted_indices[i]]; in NonMaxSuppressionSingleClassHelper()463 decoded_boxes, keep_indices[sorted_indices[i]], in NonMaxSuppressionSingleClassHelper()464 keep_indices[sorted_indices[j]]); in NonMaxSuppressionSingleClassHelper()521 int* sorted_indices = reinterpret_cast<int*>( in NonMaxSuppressionMultiClassRegularHelper() local554 output_index, num_indices_to_sort, sorted_indices); in NonMaxSuppressionMultiClassRegularHelper()558 int temp = sorted_indices[row]; in NonMaxSuppressionMultiClassRegularHelper()559 sorted_indices[row] = box_indices_after_regular_non_max_suppression[temp]; in NonMaxSuppressionMultiClassRegularHelper()564 box_indices_after_regular_non_max_suppression[row] = sorted_indices[row]; in NonMaxSuppressionMultiClassRegularHelper()
89 Tensor sorted_indices(tensorflow::DT_INT32, {how_many_labels}); in GetTopLabels() local92 sorted_indices.flat<int>()(i) = scores[i].first; in GetTopLabels()95 *out_indices = sorted_indices; in GetTopLabels()
Completed in 5 milliseconds