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 callback(self, pre, post, node_map):
      post_nms_topk = post.attrs.end[0].value
      return self.rewrite_batch_nms_with_max_out_size(
          node_map[self.cond][0],
          node_map[self.true_branch][0],
          node_map[self.data][0],
          node_map[self.valid_count][0],
          node_map[self.indices][0],
          node_map[self.iou_threshold][0],
          post_nms_topk,
      )
  
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 3 for calling the PostNMSTopKRewrite obj's underlying member method code with example usage: obj.callback(pre, post, node_map) and returns: self