code: def test_call(self):
      @polymorphic_function.function
      def foo(x):
          my_dict = {}
          my_dict["my_tensor"] = x["my_tensor"]
          my_dict["my_resource"] = x["my_variable"].handle
          my_dict["my_ints"] = x["my_ints"]
          return my_dict
  
      structured_inputs = {
          "my_tensor": constant_op.constant(1),
          "my_variable": resource_variable_ops.ResourceVariable(1),
          "my_ints": [1, 2, 3],
      }
  
      function_def, function_type = get_function_def_and_type(
          foo, (structured_inputs,)
      )
  
      atomic = atomic_function.from_function_def(
          function_def, function_type
      )
  
      result = atomic(structured_inputs)
      self.assertEqual(
          result["my_tensor"].numpy(),
          structured_inputs["my_tensor"].numpy(),
      )
      self.assertEqual(result["my_resource"].dtype, dtypes.resource)
      self.assertEqual(result["my_ints"][0].numpy(), 1)
      self.assertEqual(result["my_ints"][1].numpy(), 2)
      self.assertEqual(result["my_ints"][2].numpy(), 3)
  
file_path: tensorflow-upstream/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 6 for calling the AtomicFunctionTest obj's underlying member method code with example usage: obj.test_call() and returns: my_dict