code: def testLargeDims(self):
      # Make sure that we properly handle large inputs. See
      # https://github.com/tensorflow/tensorflow/issues/4425 for details
      for dims in [129, 256]:
          ones = np.random.rand(dims, dims).astype(np.float32)
          np_softmax = self._npSoftmax(ones)
  
          for use_gpu in [True, False]:
              with self.cached_session(use_gpu=use_gpu):
                  x = constant_op.constant(ones)
                  y = nn_ops.softmax(x)
                  tf_softmax = self.evaluate(y)
              self.assertAllClose(tf_softmax, np_softmax)
  
imports: import unittest
  import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors_impl
  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
  from tensorflow.python.platform import tf_logging
label_desc: Use image node_id 25 for calling the SoftmaxTest obj's underlying member method code with example usage: obj.testLargeDims() without return types