code: def _supported_types(self):
      if test.is_gpu_available():
          return [
              dtypes.float16,
              dtypes.bfloat16,
              dtypes.float32,
              dtypes.float64,
              dtypes.complex64,
              dtypes.complex128,
              dtypes.int64,
          ]
      return [
          dtypes.int8,
          dtypes.int16,
          dtypes.int32,
          dtypes.int64,
          dtypes.bfloat16,
          dtypes.float16,
          dtypes.float32,
          dtypes.float64,
          dtypes.complex64,
          dtypes.complex128,
      ]
  
file_path: tensorflow-upstream/tensorflow/python/kernel_tests/math_ops/aggregate_ops_test.py
imports: import numpy
  from tensorflow.core.framework import tensor_pb2
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import tensor_shape
  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 string_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the AddNTest obj's underlying member method code with example usage: obj._supported_types() and returns: list, list