def _format_input_map_as_tensors(self, input_map):
    """Returns a map from string to `tf.Tensor` or `CompositeTensor`."""
    result = {}
    for key, value in input_map.items():
      if isinstance(value, (tf.Tensor, composite_tensor.CompositeTensor)):
        result[key] = value
      else:
        result[key] = tf.convert_to_tensor(value)
    return result
Returns a map from string to `tf.Tensor` or `CompositeTensor`.
tensorflow_transform/saved/saved_transform_io_v2.py