Home
last modified time | relevance | path

Searched defs:max_size_per_class (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cc446 int num_boxes, const int max_size_per_class, const int total_size_per_batch, in BatchedNonMaxSuppressionOp()
940 const int max_size_per_class = max_output_size.scalar<int>()(); in Compute() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc6422 int max_size_per_class = in ConvertCombinedNMS() local