(record, name_to_features)
return example
Decodes a record to a TensorFlow example.
Decodes a record to a TensorFlow example.
Decodes a record to a TensorFlow example .
def _decode_record(record, name_to_features):
  """Decodes a record to a TensorFlow example."""
  example = tf.parse_single_example(record, name_to_features)

  # tf.Example only supports tf.int64, but the TPU only supports tf.int32.
  # So cast all int64 to int32.
  for name in list(example.keys()):
    t = example[name]
    if t.dtype == tf.int64:
      t = tf.to_int32(t)
    example[name] = t

  return example
def _decode_record ( record , name_to_features ) : example = tf . parse_single_example ( record , name_to_features ) # tf.Example only supports tf.int64, but the TPU only supports tf.int32. # So cast all int64 to int32. for name in list ( example . keys ( ) ) : t = example [ name ] if t . dtype == tf . int64 : t = tf . to_int32 ( t ) example [ name ] = t return example