Home
last modified time | relevance | path

Searched refs:max_classes_per_detection (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Ddetection_postprocess.cc88 int max_classes_per_detection; // Fast Non-Max-Suppression member
106 op_data->max_classes_per_detection = m["max_classes_per_detection"].AsInt32(); in Init()
165 op_data->max_detections * op_data->max_classes_per_detection; in Prepare()
677 const int max_categories_per_anchor = op_data->max_classes_per_detection; in NonMaxSuppressionMultiClassFastHelper()
/external/tensorflow/tensorflow/lite/micro/kernels/
Ddetection_postprocess.cc92 int max_classes_per_detection; // Fast Non-Max-Suppression member
130 op_data->max_classes_per_detection = m["max_classes_per_detection"].AsInt32(); in Init()
637 const int max_categories_per_anchor = op_data->max_classes_per_detection; in NonMaxSuppressionMultiClassFastHelper()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dprepare-composite-functions-tf.mlir551 …nc<@"TFLite_Detection_PostProcess", {max_detections = 10 : i64, max_classes_per_detection = 1 : i6…
572 …nc<@"TFLite_Detection_PostProcess", {max_detections = 10 : i64, max_classes_per_detection = 1 : i6…
575 …nc<@"TFLite_Detection_PostProcess", {max_detections = 10 : i64, max_classes_per_detection = 1 : i6…
577 // expected-error @+1 {{max_classes_per_detection attribute is not set or not an integer}}
Dops.mlir617 …r_class = 100 : i64, device = "", h_scale = 5.000000e+00 : f32, max_classes_per_detection = 1 : i6…