Searched defs:iou_threshold (Results 1 – 4 of 4) sorted by relevance
108 typename TTypes<T, 2>::ConstTensor boxes, int i, int j, T iou_threshold) { in IOUGreaterThanThreshold() argument229 const float score_threshold, const float iou_threshold, in BatchedNonMaxSuppressionOp()496 const Tensor& iou_threshold = context->input(3); in Compute() local537 const Tensor& iou_threshold = context->input(3); in Compute() local588 const Tensor& iou_threshold = context->input(3); in Compute() local710 const Tensor& iou_threshold = context->input(4); in Compute() local
34 void MakeOp(float iou_threshold) { in MakeOp()
119 ShapeHandle iou_threshold; in NMSShapeFn() local814 ShapeHandle iou_threshold; in __anonc970e8b61702() local
2444 iou_threshold=0.5, argument2496 iou_threshold=0.5, argument3483 iou_threshold=0.5, argument