Searched refs:_TFRecordDataset (Results 1 – 2 of 2) sorted by relevance
38 def _TFRecordDataset(filename): function45 'tfrecord': _TFRecordDataset,
88 class _TFRecordDataset(dataset_ops.DatasetSource): class115 super(_TFRecordDataset, self).__init__(variant_tensor)219 return _TFRecordDataset(filename, compression_type, buffer_size)