code: def test_layout_prop_v2_with_const_tf_function(self):
      a = constant_op.constant([[1.0, 2.0], [3.0, 4.0]])
      b = constant_op.constant([[10.0, 20.0], [30.0, 40.0]])
      golden_result = math_ops.add(a, b)
  
      c = api.copy_to_mesh(a, self.unsharded_unsharded_layout)
  
      @polymorphic_function.function
      def add_function():
          d = constant_op.constant([[10.0, 20.0], [30.0, 40.0]])
          return math_ops.add(c, d)
  
      dtensor_result = add_function()
      self.assertDTensorEqual(
          golden_result, self.unsharded_unsharded_layout, dtensor_result
      )
  
imports: from absl.testing import parameterized
  import numpy
  from tensorflow.dtensor.python import api
  from tensorflow.dtensor.python import d_variable
  from tensorflow.dtensor.python import layout
  from tensorflow.dtensor.python import numpy_util
  from tensorflow.dtensor.python.tests import test_util
  from tensorflow.python.eager import backprop
  from tensorflow.python.eager.polymorphic_function import polymorphic_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gen_nn_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops import stateless_random_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the LayoutPropagationV2Test obj's underlying member method code with example usage: obj.test_layout_prop_v2_with_const_tf_function() and returns: math_ops