docstr: Ensure that the correct elements are returned.
docstr_tok: ["Ensure","that","the","correct","elements","are","returned","."]
code: def test_load_saved_model_graph_with_return_elements(
      self, builder_cls
  ):
      """Ensure that the correct elements are returned."""
      self.export_simple_graph(builder_cls)
      loader = loader_impl.SavedModelLoader(SIMPLE_ADD_SAVED_MODEL)
      graph = ops.Graph()
      _, ret = loader.load_graph(
          graph, ["foo_graph"], return_elements=["y:0", "x:0"]
      )
  
      self.assertEqual(graph.get_tensor_by_name("y:0"), ret[0])
      self.assertEqual(graph.get_tensor_by_name("x:0"), ret[1])
  
      with self.assertRaisesRegex(ValueError, "not found in graph"):
          loader.load_graph(
              graph, ["foo_graph"], return_elements=["z:0"]
          )
  
imports: import os
  import shutil
  from absl.testing import parameterized
  from tensorflow.python.client import session
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import control_flow_ops
  from tensorflow.python.ops import resource_variables_toggle
  from tensorflow.python.ops import state_ops
  from tensorflow.python.ops import variable_v1
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
  from tensorflow.python.saved_model import builder
  from tensorflow.python.saved_model import loader_impl
  from tensorflow.python.saved_model import signature_def_utils
  from tensorflow.python.saved_model import utils
  from tensorflow.python.training import saver
label_desc: Use image node_id 12 for calling the SavedModelLoaderTest obj's underlying member method code with example usage: obj.test_load_saved_model_graph_with_return_elements(builder_cls) without return types