Searched refs:d_selected_indices (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.h | 55 const float iou_threshold, int* d_selected_indices,
|
| D | non_max_suppression_op.cu.cc | 253 const float iou_threshold, int* d_selected_indices, int* h_nkeep, in NmsGpu() argument
350 d_selected_indices, // selected items in NmsGpu()
449 Tensor d_selected_indices; in DoNMS() local
451 DataType::DT_INT32, TensorShape({num_boxes}), &d_selected_indices)); in DoNMS()
512 iou_threshold_val, d_selected_indices.flat<int>().data(), in DoNMS()
542 d_selected_indices.flat<int>().data(), sorted_indices, in DoNMS()
|