code: def test_conv2d(self):
      x = array_ops.fill((16, 40, 40, 64), 1 + 2**-12)
      y = array_ops.fill((3, 3, 64, 64), 1.0)
  
      def conv2d(x, y):
          return nn_ops.conv2d(x, y, [1, 1, 1, 1], padding="SAME")
  
      self._test_fn(conv2d, [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 5 for calling the TensorFloat32Test obj's underlying member method code with example usage: obj.test_conv2d() and returns: nn_ops