code: def topk_and_argtopk(x, kth, axis=-1, sorted=True, idx_dtype="int64"):
      """
      Returns the results of both topk() and argtopk() in one Op.
  
      See the respective documentation for details.
  
      Returns
      -------
      tuple: (values, indices)
  
      """
      if axis is None:
          x = flatten(x)
          axis = 0
      return TopKOp(axis=axis, sorted=sorted, idx_dtype=idx_dtype)(
          x, kth
      )
  
imports: import numpy
  from pytensor.gradient import grad_undefined
  from pytensor.graph.basic import Apply, Constant
  from pytensor.graph.op import Op
  from pytensor.misc.safe_asarray import _asarray
  from pytensor.tensor.basic import arange, as_tensor_variable, flatten, switch
  from pytensor.tensor.math import eq, ge, mul
  from pytensor.tensor.shape import shape
  from pytensor.tensor.subtensor import set_subtensor
  from pytensor.tensor.type import TensorType, integer_dtypes
label_desc: Use image node_id 8 for calling a global function with example usage: topk_and_argtopk(x, kth, axis, sorted, idx_dtype) and returns: TopKOp