Searched refs:final_valid_detections (Results 1 – 1 of 1) sorted by relevance
409 std::vector<int>& final_valid_detections, in SelectResultPerBatch() argument429 final_valid_detections[batch_idx] = max_detections; in SelectResultPerBatch()492 std::vector<int> final_valid_detections(num_batches); in BatchedNonMaxSuppressionOp() local537 final_valid_detections, batch_idx, total_size_per_batch, in BatchedNonMaxSuppressionOp()540 valid_detections_flat(batch_idx) = final_valid_detections[batch_idx]; in BatchedNonMaxSuppressionOp()