Searched refs:max_size_per_class (Results 1 – 2 of 2) sorted by relevance
228 int num_boxes, const int max_size_per_class, const int total_size_per_batch, in BatchedNonMaxSuppressionOp() argument296 const int size_per_class = std::min(max_size_per_class, num_boxes); in BatchedNonMaxSuppressionOp()365 std::min(total_size_per_batch, max_size_per_class * num_classes); in BatchedNonMaxSuppressionOp()699 const int max_size_per_class = max_output_size.scalar<int>()(); in Compute() local735 max_size_per_class, max_total_size_per_batch, in Compute()
51 `max_size_per_class`*`num_classes`, unless it exceeds `max_total_size` in