code: def nms_shape_func(attrs, inputs, _):
      if attrs.return_indices:
          return _nms_shape_func(inputs[0])
      return [topi.math.identity(inputs[0])]
  
file_path: tvm/python/tvm/relay/op/vision/_vision.py
imports: from __future__ import absolute_import
  from tvm import topi
  from tvm.te.hybrid import script
  from tvm.runtime import convert
  from ..None import op
  from ..None import strategy
  from ..op import OpPattern
label_desc: Use image node_id 4 for calling a global function with example usage: nms_shape_func(attrs, inputs, _) and returns: list, _nms_shape_func