Method _testLargeSort

tensorflow/python/kernel_tests/topk_op_test.py:115–122

(self, dtype)

    self._validateTopK(inputs, 1, [[np.NaN], [np.NaN]], [[0], [0]])

  def _testLargeSort(self, dtype):
    b = 10
    n = 5000
    inputs = np.random.permutation(
        np.linspace(0, 100, b * n, dtype=dtype)).reshape(b, n)
    indices = np.argsort(-inputs, axis=1)
    values = -np.sort(-inputs, axis=1)
    self._validateTopK(inputs, n, values, indices)

  def testLargeSort(self):
    self._testLargeSort(np.float32)