class_docstr: Test cases for segment reduction ops.
code: def testSegmentMaxNumSegmentsMore(self):
      for dtype in self.int_types | self.float_types:
          minval = dtypes.as_dtype(dtype).min
          if dtype == np.float64 and self._findDevice("TPU"):
              minval = -np.Inf
          self.assertAllClose(
              np.array([1, minval, 2, 5, minval], dtype=dtype),
              self._segmentMaxV2(
                  np.array([0, 1, 2, 3, 4, 5], dtype=dtype),
                  np.array([0, 0, 2, 3, 3, 3], dtype=np.int32),
                  5,
              ),
          )
  
imports: import functools
  import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.python.client import device_lib
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import googletest
label_desc: Use image node_id 20 for calling the SegmentReductionOpsTest obj's underlying member method code with example usage: obj.testSegmentMaxNumSegmentsMore() without return types