code: def test_nonjit(self, dtype, k, row_size, num_rows):
      # Support regular topk semantics.
      row = np.arange(row_size, dtype=np.float32)
      db = np.stack(
          list(self._rng.permutation(row) for _ in range(num_rows))
      )
      db_tensor = constant_op.constant(db, dtype=dtype)
      _, idx = self.evaluate(nn_ops.approx_max_k(db_tensor, k))
      sorted_idx = np.sort(idx)
      expected = np.sort(np.argsort(-db)[:, :k])
      self.assertAllEqual(sorted_idx, expected)
  
imports: import itertools
  from absl.testing import parameterized
  import numpy
  from tensorflow.python.eager import backprop
  from tensorflow.python.eager.def_function import function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 11 for calling the ApproxTopkTest obj's underlying member method code with example usage: obj.test_nonjit(dtype, k, row_size, num_rows) without return types