Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dnon_max_suppression.cc111 const TfLiteTensor* input_iou_threshold; in Prepare() local
114 &input_iou_threshold)); in Prepare()
115 TF_LITE_ENSURE_EQ(context, input_iou_threshold->type, kTfLiteFloat32); in Prepare()
116 TF_LITE_ENSURE_EQ(context, NumDimensions(input_iou_threshold), 0); in Prepare()
121 TF_LITE_ENSURE_EQ(context, input_iou_threshold->type, kTfLiteFloat32); in Prepare()
218 const TfLiteTensor* input_iou_threshold; in Eval() local
221 &input_iou_threshold)); in Eval()
222 const float iou_threshold = *GetTensorData<float>(input_iou_threshold); in Eval()