Searched refs:nms_op (Results 1 – 2 of 2) sorted by relevance
2551 LogicalResult matchAndRewrite(NonMaxSuppressionV3Op nms_op, in matchAndRewrite() argument2553 if (nms_op.getNumOperands() != 5) { in matchAndRewrite()2557 new_result_types.push_back(nms_op.getType()); in matchAndRewrite()2558 auto input_ty = nms_op.getType().template cast<ShapedType>(); in matchAndRewrite()2565 nms_op.getLoc(), new_result_types, nms_op.boxes(), nms_op.scores(), in matchAndRewrite()2566 nms_op.max_output_size(), nms_op.iou_threshold(), in matchAndRewrite()2567 nms_op.score_threshold()); in matchAndRewrite()2570 nms_op.replaceAllUsesWith(nmsv4.getResult(0)); in matchAndRewrite()
3149 auto nms_op = ops::CombinedNonMaxSuppression( in TEST_P() local3153 return nms_op.operation.node()->def(); in TEST_P()