def METHOD_NAME(dataset_construction_graph, return_element):
  with tf.Graph().as_default() as graph:
    v1 = tf.graph_util.import_graph_def(
        dataset_construction_graph.as_graph_def(),
        return_elements=[return_element],
    )[0]
    structure = tf.TensorSpec([], tf.int64)
    ds1 = tf.data.experimental.from_variant(v1, structure=structure)
    out = ds1.reduce(tf.constant(0, dtype=tf.int64), lambda x, y: x + y)
  return graph, '', out.name