code: def _topk_shape_func_input_data(data, k, axis):
      ndim = len(data.shape)
      val_out = output_tensor((ndim,), "int64")
      indices_out = output_tensor((ndim,), "int64")
  
      for i in const_range(ndim):
          if i != axis:
              val_out[i] = int64(data.shape[i])
              indices_out[i] = int64(data.shape[i])
          else:
              if k[0] < 1:
                  val_out[i] = int64(data.shape[i])
                  indices_out[i] = int64(data.shape[i])
              else:
                  val_out[i] = int64(k[0])
                  indices_out[i] = int64(k[0])
      return val_out, indices_out
  
file_path: tvm/python/tvm/relay/op/dyn/_algorithm.py
imports: from __future__ import absolute_import
  from tvm.te.hybrid import script
  from tvm.runtime import convert
  from ..None import strategy
  from ..None import op
  from ..op import OpPattern, register_pattern
  from ..op import register_strategy
label_desc: Use image node_id 1 for calling a global function with example usage: _topk_shape_func_input_data(data, k, axis) and returns: val_out,  indices_out