Searched defs:max_size_per_class (Results 1 – 2 of 2) sorted by relevance
466 int num_boxes, const int max_size_per_class, const int total_size_per_batch, in BatchedNonMaxSuppressionOp()981 const int max_size_per_class = max_output_size.scalar<int>()(); in Compute() local
5243 const int max_size_per_class = *(output_size_per_class.GetPointer<int>()); in ConvertCombinedNMS() local5431 int max_size_per_class = *(output_size_per_class.GetPointer<int>()); in ConvertCombinedNMS() local