Searched refs:total_size_per_batch (Results 1 – 1 of 1) sorted by relevance
228 int num_boxes, const int max_size_per_class, const int total_size_per_batch, in BatchedNonMaxSuppressionOp() argument247 int per_batch_size = total_size_per_batch; in BatchedNonMaxSuppressionOp()361 std::min((int)result_candidate_vec.size(), total_size_per_batch); in BatchedNonMaxSuppressionOp()362 per_batch_size = total_size_per_batch; in BatchedNonMaxSuppressionOp()365 std::min(total_size_per_batch, max_size_per_class * num_classes); in BatchedNonMaxSuppressionOp()