Searched refs:iou_thresh (Results 1 – 2 of 2) sorted by relevance
445 const xla::XlaOp iou_thresh = context->Input("iou_threshold"); in Compile() local515 xla::XlaOp iou_thresh_mask = xla::Gt(iou, iou_thresh + square_zero); in Compile()
5983 float iou_thresh = *(static_cast<float*>(iou_threshold.GetValues())); in ConvertCombinedNMS() local5984 if (iou_thresh < 0.0 || iou_thresh > 1.0) { in ConvertCombinedNMS()6035 nvinfer1::PluginField{"iouThreshold", &iou_thresh, in ConvertCombinedNMS()