Searched refs:max_size_per_class (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cc | 443 int num_boxes, const int max_size_per_class, const int total_size_per_batch, in BatchedNonMaxSuppressionOp() argument
456 const int size_per_class = std::min(max_size_per_class, num_boxes); in BatchedNonMaxSuppressionOp()
500 std::min(total_size_per_batch, max_size_per_class * num_classes); in BatchedNonMaxSuppressionOp()
515 pad_per_class, max_size_per_class * num_classes, clip_boxes, in BatchedNonMaxSuppressionOp()
931 const int max_size_per_class = max_output_size.scalar<int>()(); in Compute() local
973 max_size_per_class, max_total_size_per_batch, in Compute()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CombinedNonMaxSuppression.pbtxt | 53 `max_size_per_class`*`num_classes`, unless it exceeds `max_total_size` in
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 5960 int max_size_per_class = in ConvertCombinedNMS() local
5962 if (max_size_per_class <= 0) { in ConvertCombinedNMS()
6012 keep_top_k = std::min(max_size_per_class * num_classes, max_total_size); in ConvertCombinedNMS()
|