Home
last modified time | relevance | path

Searched refs:sorted_indices (Results 1 – 10 of 10) sorted by relevance

/external/protobuf/src/google/protobuf/
Dgenerated_enum_util.cc70 int LookUpEnumName(const EnumEntry* enums, const int* sorted_indices, in LookUpEnumName() argument
76 std::lower_bound(sorted_indices, sorted_indices + size, -1, comparator); in LookUpEnumName()
77 if (it != sorted_indices + size && enums[*it].value == value) { in LookUpEnumName()
78 return it - sorted_indices; in LookUpEnumName()
84 const EnumEntry* enums, const int* sorted_indices, size_t size, in InitializeEnumStrings() argument
87 enum_strings[i].Construct(enums[sorted_indices[i]].name); in InitializeEnumStrings()
Dgenerated_enum_util.h69 const int* sorted_indices, size_t size,
74 const EnumEntry* enums, const int* sorted_indices, size_t size,
/external/tensorflow/tensorflow/lite/kernels/
Ddetection_postprocess.cc452 std::vector<int> sorted_indices; in NonMaxSuppressionSingleClassHelper() local
453 sorted_indices.resize(num_scores_kept); in NonMaxSuppressionSingleClassHelper()
455 sorted_indices.data()); in NonMaxSuppressionSingleClassHelper()
471 selected->push_back(keep_indices[sorted_indices[i]]); in NonMaxSuppressionSingleClassHelper()
481 decoded_boxes, keep_indices[sorted_indices[i]], in NonMaxSuppressionSingleClassHelper()
482 keep_indices[sorted_indices[j]]); in NonMaxSuppressionSingleClassHelper()
552 std::vector<int> sorted_indices; in NonMaxSuppressionMultiClassRegularHelper() local
553 sorted_indices.resize(num_boxes + max_detections); in NonMaxSuppressionMultiClassRegularHelper()
582 sorted_indices.data()); in NonMaxSuppressionMultiClassRegularHelper()
586 int temp = sorted_indices[row]; in NonMaxSuppressionMultiClassRegularHelper()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Ddetection_postprocess.cc434 int* sorted_indices = reinterpret_cast<int*>( in NonMaxSuppressionSingleClassHelper() local
438 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() local
554 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()
/external/libtextclassifier/native/utils/container/
Dbit-vector_test.cc55 std::vector<int32> sorted_indices = {3, 7}; in TEST() local
58 BitVector::CreateSparseBitVectorData(sorted_indices); in TEST()
/external/tensorflow/tensorflow/examples/speech_commands/
Dlabel_wav.cc89 Tensor sorted_indices(tensorflow::DT_INT32, {how_many_labels}); in GetTopLabels() local
92 sorted_indices.flat<int>()(i) = scores[i].first; in GetTopLabels()
95 *out_indices = sorted_indices; in GetTopLabels()
/external/tensorflow/tensorflow/compiler/tests/
Dsort_ops_test.py147 sorted_indices = np.argsort(inputs[0], axis=dimension)
149 np.take_along_axis(inp, sorted_indices, axis=dimension)
189 sorted_indices = np.argsort(100 * inputs[0] + inputs[1])
191 np.take_along_axis(inp, sorted_indices, axis=0) for inp in inputs
/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cu.cc481 const int* sorted_indices = d_sorted_indices.flat<int>().data(); in DoNMS() local
485 config.virtual_thread_count, sorted_indices, in DoNMS()
542 d_selected_indices.flat<int>().data(), sorted_indices, in DoNMS()
/external/tensorflow/tensorflow/python/ops/linalg/sparse/
Dsparse_csr_matrix_ops.py305 def sorted_indices(self): member in SparseMatrix
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_impl.py5491 scores, boxes, sorted_indices = _sort_scores_and_boxes(scores, boxes)
5494 sorted_indices = array_ops.zeros_like(scores, dtype=dtypes.int32)
5547 array_ops.gather(array_ops.reshape(sorted_indices, [-1]),