class_docstr: A callback to rewrite nms to exploit max_out_size parameter.
class_docstr_tok: ["A","callback","to","rewrite","nms","to","exploit","max_out_size","parameter","."]
code: def __init__(self):
      super().__init__()
      self.cond = wildcard()
      self.true_branch = wildcard()
      self.data = wildcard()
      self.valid_count = wildcard()
      self.indices = wildcard()
      self.iou_threshold = wildcard()
  
      self.pattern = topk_after_batch_nms_pattern(
          self.cond,
          self.true_branch,
          self.data,
          self.valid_count,
          self.indices,
          self.iou_threshold,
      )
  
imports: from ..None import expr
  from ..None import op
  from ..dataflow_pattern import wildcard, is_constant, is_op, rewrite, is_tuple, is_tuple_get_item, is_if, DFPatternCallback
label_desc: Use image node_id 1 to create a new PostNMSTopKRewrite object from inherited base classes: DFPatternCallback with example: obj = PostNMSTopKRewrite()