Searched refs:size_per_class (Results 1 – 2 of 2) sorted by relevance
306 int num_classes, const int size_per_class, in DoNMSPerClass() argument352 while (selected.size() < size_per_class && in DoNMSPerClass()371 result_candidate_vec[selected.size() + size_per_class * class_idx] = { in DoNMSPerClass()456 const int size_per_class = std::min(max_size_per_class, num_boxes); in BatchedNonMaxSuppressionOp() local459 std::vector<ResultCandidate>(size_per_class * num_classes, in BatchedNonMaxSuppressionOp()478 num_classes, size_per_class, score_threshold, iou_threshold, in BatchedNonMaxSuppressionOp()
234 DimensionHandle size_per_class; in CombinedNMSShapeFn() local235 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(2, &size_per_class)); in CombinedNMSShapeFn()243 if (c->ValueKnown(size_per_class) && c->Value(size_per_class) <= 0) { in CombinedNMSShapeFn()249 c->Value(size_per_class) * c->Value(class_dim)); in CombinedNMSShapeFn()