message: Use optimized ArgMax to replace partial ArgSort when the number of category is one.
diff: @@ -364,11 +364,15 @@ TfLiteStatus DecodeCenterSizeBoxes(TfLiteContext* context, TfLiteNode* node,
  void DecreasingPartialArgSort(const float* values, int num_values,
  int num_to_sort, int* indices) {
  + if (num_to_sort == 1) {
  + indices[0] = optimized_ops::ArgMaxVector(values, num_values);
  + } else {
  std::iota(indices, indices + num_values, 0);
  std::partial_sort(
  indices, indices + num_to_sort, indices + num_values,
  [&values](const int i, const int j) { return values[i] > values[j]; });
  }
  +}
  void DecreasingArgSort(const float* values, int num_values, int* indices) {
  std::iota(indices, indices + num_values, 0);
  