code: def test_batch_matmul(self):
      x = array_ops.fill((2, 1024, 1024), 1 + 2**-12)
      y = array_ops.fill((2, 1024, 1024), 1.0)
  
      def batch_matmul(x, y):
          return math_ops.matmul(x, y)
  
      self._test_fn(batch_matmul, [x, y])
  
file_path: tensorflow-upstream/tensorflow/compiler/tests/tensor_float_32_test.py
imports: from tensorflow.compiler.tests import xla_test
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import config
  from tensorflow.python.framework import ops
  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 nn_ops
  from tensorflow.python.platform import googletest
  from tensorflow.python.platform import sysconfig
label_desc: Use image node_id 4 for calling the TensorFloat32Test obj's underlying member method code with example usage: obj.test_batch_matmul() and returns: math_ops