code: def testAdd(self):
      if xla_test.test.is_built_with_rocm():
          self.skipTest("Broken with rocm")
      for dtype in self.numeric_types:
          self._assertOpOutputMatchesExpected(
              xla.add,
              args=(
                  np.array([1, 2, 3], dtype=dtype),
                  np.array([4, 5, 6], dtype=dtype),
              ),
              expected=np.array([5, 7, 9], dtype=dtype),
          )
  
          self._assertOpOutputMatchesExpected(
              lambda x, y: xla.add(x, y, broadcast_dims=(0,)),
              args=(
                  np.array([[1, 2], [3, 4]], dtype=dtype),
                  np.array([7, 11], dtype=dtype),
              ),
              expected=np.array([[8, 9], [14, 15]], dtype=dtype),
          )
  
          self._assertOpOutputMatchesExpected(
              lambda x, y: xla.add(x, y, broadcast_dims=(1,)),
              args=(
                  np.array([[1, 2], [3, 4]], dtype=dtype),
                  np.array([7, 11], dtype=dtype),
              ),
              expected=np.array([[8, 13], [10, 15]], dtype=dtype),
          )
  
imports: import functools
  from absl.testing import parameterized
  import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.compiler.tf2xla.ops import gen_xla_ops
  from tensorflow.compiler.tf2xla.python import xla
  from local_xla.xla import xla_data_pb2
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import function
  from tensorflow.python.framework import ops
  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 array_ops_stack
  from tensorflow.python.ops import random_ops_util
  from tensorflow.python.platform import googletest
label_desc: Use image node_id 2 for calling the XlaOpsNumericalTest obj's underlying member method code with example usage: obj.testAdd() without return types