Searched refs:size_per_class (Results 1 – 2 of 2) sorted by relevance
174 DimensionHandle size_per_class; in CombinedNMSShapeFn() local175 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(2, &size_per_class)); in CombinedNMSShapeFn()183 if (c->ValueKnown(size_per_class) && c->Value(size_per_class) <= 0) { in CombinedNMSShapeFn()189 c->Value(size_per_class) * c->Value(class_dim)); in CombinedNMSShapeFn()
296 const int size_per_class = std::min(max_size_per_class, num_boxes); in BatchedNonMaxSuppressionOp() local322 while (selected.size() < size_per_class && in BatchedNonMaxSuppressionOp()