def METHOD_NAME():
  """Builds the schema of the data from TFRecord to be inputted in the model.
  Returns:
    A dictionary mapping keys of column names to `tf.FixedLenFeature` and
    specifies the shape and dtypes.
  """
  schema = {}
  for c in constants.FEATURE_COLUMNS:
    schema[c] = tf.FixedLenFeature(shape=[], dtype=tf.float32)
  schema[constants.LABEL_COLUMN] = tf.FixedLenFeature(shape=[], dtype=tf.int64)
  schema[constants.KEY_COLUMN] = tf.FixedLenFeature(shape=[], dtype=tf.int64)
  return schema