Searched refs:max_boxes (Results 1 – 2 of 2) sorted by relevance
57 const int max_boxes, bool flip_boxes = false);
118 const int num_boxes, const int max_boxes, in NMSReduce() argument139 if (accepted_boxes > max_boxes) break; in NMSReduce()254 OpKernelContext* context, const int max_boxes, bool flip_boxes) { in NmsGpu() argument326 num_boxes, max_boxes, selected)); in NmsGpu()