Searched defs:num_boxes_to_generate (Results 1 – 1 of 1) sorted by relevance
190 int num_boxes_to_generate, int box_dim) { in AllocateGenerationTempTensors()226 Tensor* dev_prenms_nboxes, int num_images, int num_boxes_to_generate, in AllocatePreNMSTempTensors()