Home
last modified time | relevance | path

Searched refs:NmsGpu (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.h54 Status NmsGpu(const float* d_sorted_boxes_float_ptr, const int num_boxes,
Dnon_max_suppression_op.cu.cc252 Status NmsGpu(const float* d_sorted_boxes_float_ptr, const int num_boxes, in NmsGpu() function
511 auto status = NmsGpu(d_sorted_boxes.flat<float>().data(), limited_num_boxes, in DoNMS()
Dgenerate_box_proposals_op.cu.cc515 OP_REQUIRES_OK(context, NmsGpu(d_image_prenms_boxes, prenms_nboxes, in Compute()