Searched refs:iou_threshold_ (Results 1 – 1 of 1) sorted by relevance
443 OP_REQUIRES_OK(context, context->GetAttr("iou_threshold", &iou_threshold_)); in NonMaxSuppressionOp()458 OP_REQUIRES(context, iou_threshold_ >= 0 && iou_threshold_ <= 1, in Compute()467 CreateIOUSuppressCheckFn<float>(boxes, iou_threshold_); in Compute()475 float iou_threshold_; member in tensorflow::NonMaxSuppressionOp