Searched defs:size_per_class (Results 1 – 2 of 2) sorted by relevance
341 int num_classes, const int size_per_class, in DoNMSPerClass()479 const int size_per_class = std::min(max_size_per_class, num_boxes); in BatchedNonMaxSuppressionOp() local
234 DimensionHandle size_per_class; in CombinedNMSShapeFn() local