def _get_tensor_value_to_key_map(features_dict):
  """Get reverse map from name of tensor values to key in `features_dict`."""
  result = {}
  for key, tensor in features_dict.items():
    values = _get_tensor_values(tensor)
    result[values.name] = key
  return result
Get reverse map from name of tensor values to key in `features_dict`.
tensorflow_transform/schema_inference.py