def _update_cache_value_node_references(cache_value_nodes, traverser):
  """Updates value node references in the cache."""
  if cache_value_nodes:
    cache_value_nodes = {
        key: traverser.visit_value_node(value_node)
        for key, value_node in cache_value_nodes.items()
    }
  return cache_value_nodes
Updates value node references in the cache.
tensorflow_transform/beam/combiner_packing_util.py