code: def test_call_graph(self):
      definition, func_type = get_function_def_and_type(
          lambda x, y: x + y,
          (constant_op.constant(1), constant_op.constant(2)),
      )
  
      atomic = atomic_function.from_function_def(definition, func_type)
  
      @polymorphic_function.function
      def foo(a, b):
          return atomic.call_flat(a, b)[0]
  
      self.assertEqual(
          foo(constant_op.constant(3), constant_op.constant(4)).numpy(),
          7,
      )
  
file_path: tensorflow/tensorflow/python/eager/polymorphic_function/atomic_function_test.py
imports: from tensorflow.python.eager.polymorphic_function import atomic_function
  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.framework import ops
  from tensorflow.python.ops import resource_variable_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the AtomicFunctionTest obj's underlying member method code with example usage: obj.test_call_graph() and returns: atomic