code: def argtopk(
      a,
      k,
      axis=-1,
      largest=True,
      sorted=True,
      order=None,
      parallel_kind="auto",
      psrs_kinds=None,
  ):
      (
          a,
          k,
          axis,
          largest,
          sorted,
          order,
          parallel_kind,
          psrs_kinds,
      ) = _validate_topk_arguments(
          a, k, axis, largest, sorted, order, parallel_kind, psrs_kinds
      )
      op = TensorTopk(
          k=k,
          axis=axis,
          largest=largest,
          sorted=sorted,
          parallel_kind=parallel_kind,
          psrs_kinds=psrs_kinds,
          dtype=a.dtype,
          return_value=False,
          return_indices=True,
          stage=OperandStage.agg,
      )
      return op(a)
  
label_desc: Use image node_id 1 for calling a global function with example usage: argtopk(a, k, axis, largest, sorted, order, parallel_kind, psrs_kinds) and returns: op