Searched refs:keep_top_k (Results 1 – 1 of 1) sorted by relevance
6010 int keep_top_k = 0; in ConvertCombinedNMS() local6012 keep_top_k = std::min(max_size_per_class * num_classes, max_total_size); in ConvertCombinedNMS()6014 keep_top_k = max_total_size; in ConvertCombinedNMS()6019 const int top_k = std::max(num_boxes, keep_top_k); in ConvertCombinedNMS()6031 nvinfer1::PluginField{"keepTopK", &keep_top_k, in ConvertCombinedNMS()