Searched refs:nms_op (Results 1 – 2 of 2) sorted by relevance
2744 LogicalResult matchAndRewrite(NonMaxSuppressionV3Op nms_op, in matchAndRewrite() argument2746 if (nms_op.getNumOperands() != 5) { in matchAndRewrite()2750 new_result_types.push_back(nms_op.getType()); in matchAndRewrite()2751 auto input_ty = nms_op.getType().template cast<ShapedType>(); in matchAndRewrite()2758 nms_op.getLoc(), new_result_types, nms_op.boxes(), nms_op.scores(), in matchAndRewrite()2759 nms_op.max_output_size(), nms_op.iou_threshold(), in matchAndRewrite()2760 nms_op.score_threshold()); in matchAndRewrite()2763 nms_op.replaceAllUsesWith(nmsv4.getResult(0)); in matchAndRewrite()
4380 auto nms_op = ops::CombinedNonMaxSuppression( in TEST_P() local4384 return nms_op.operation.node()->def(); in TEST_P()4580 auto nms_op = ops::CombinedNonMaxSuppression( in TEST_P() local4584 return nms_op.operation.node()->def(); in TEST_P()