    def __init__(self, iou_threshold=0.5, keep_n=-1) -> None:
        """Op Initialization"""
        super().__init__()
        self._attrs["op"] = "batched_nms"
        self._attrs["has_profiler"] = False
        self._attrs["keep_n"] = keep_n
        self._attrs["iou_threshold"] = iou_threshold
        self.exec_key_template = EXEC_KEY_TEMPLATE
Op Initialization
python/aitemplate/compiler/ops/vision_ops/nms/batched_nms.py