Searched refs:CustomReader (Results 1 – 2 of 2) sorted by relevance
48 CustomReader::kSnappyReaderInputBufferSizeBytes;50 CustomReader::kSnappyReaderOutputBufferSizeBytes;331 *out_reader = absl::make_unique<CustomReader>(filename, compression_type, in Create()651 CustomReader::CustomReader(const std::string& filename, in CustomReader() function in tensorflow::data::snapshot_util::CustomReader659 Status CustomReader::Initialize(Env* env) { in Initialize()701 Status CustomReader::ReadTensors(std::vector<Tensor>* read_tensors) { in ReadTensors()755 Status CustomReader::ReadTensorsV0(std::vector<Tensor>* read_tensors) { in ReadTensorsV0()776 Status CustomReader::SnappyUncompress( in SnappyUncompress()826 Status CustomReader::ReadRecord(tstring* record) { in ReadRecord()834 Status CustomReader::ReadRecord(absl::Cord* record) { in ReadRecord()
230 class CustomReader : public Reader {247 CustomReader(const std::string& filename, const string& compression_type,252 ~CustomReader() override {} in ~CustomReader()