Home
last modified time | relevance | path

Searched refs:input_stream_ (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/lib/io/
Drecord_reader.cc60 input_stream_(new RandomAccessInputStream(file)), in RecordReader()
63 input_stream_.reset(new BufferedInputStream(input_stream_.release(), in RecordReader()
72 input_stream_.reset(new ZlibInputStream( in RecordReader()
73 input_stream_.release(), options.zlib_options.input_buffer_size, in RecordReader()
77 input_stream_.reset( in RecordReader()
78 new SnappyInputStream(input_stream_.release(), in RecordReader()
99 TF_RETURN_IF_ERROR(input_stream_->ReadNBytes(expected, result)); in ReadChecksummed()
125 TF_RETURN_IF_ERROR(input_stream_->Reset()); in GetMetadata()
151 TF_RETURN_IF_ERROR(input_stream_->SkipNBytes(length + kFooterSize)); in GetMetadata()
171 int64 curr_pos = input_stream_->Tell(); in PositionInputStream()
[all …]
Dbuffered_inputstream.cc26 : input_stream_(input_stream), in BufferedInputStream()
39 delete input_stream_; in ~BufferedInputStream()
49 Status s = input_stream_->ReadNBytes(size_, &buf_); in FillBuffer()
139 Status s = input_stream_->SkipNBytes(bytes_to_skip - (limit_ - pos_)); in SkipNBytes()
151 return input_stream_->Tell() - (limit_ - pos_); in Tell()
161 const int64 buf_lower_limit = input_stream_->Tell() - limit_; in Seek()
202 TF_RETURN_IF_ERROR(input_stream_->Reset()); in Reset()
Dzlib_inputstream.cc61 input_stream_(input_stream), in ZlibInputStream()
83 delete input_stream_; in ~ZlibInputStream()
91 TF_RETURN_IF_ERROR(input_stream_->Reset()); in Reset()
146 Status s = input_stream_->ReadNBytes(bytes_to_read, &data); in ReadFromStream()
Dzlib_inputstream.h83 InputStreamInterface* input_stream_; variable
Dbuffered_inputstream.h93 InputStreamInterface* input_stream_; // not owned. variable
Drecord_reader.h123 std::unique_ptr<InputStreamInterface> input_stream_; variable
/external/tensorflow/tensorflow/core/lib/io/snappy/
Dsnappy_inputstream.cc28 : input_stream_(input_stream), in SnappyInputStream()
42 delete input_stream_; in ~SnappyInputStream()
87 input_stream_->ReadNBytes(sizeof(uint32), &compressed_block_length_ts)); in Inflate()
98 input_stream_->ReadNBytes(compressed_block_length, &compressed_block); in Inflate()
146 TF_RETURN_IF_ERROR(input_stream_->Reset()); in Reset()
Dsnappy_inputstream.h65 InputStreamInterface* input_stream_; variable
/external/tensorflow/tensorflow/core/kernels/data/
Dtext_line_dataset_op.cc194 input_stream_ = in SetupStreamsLocked()
199 input_stream_.get(), dataset()->options_.input_buffer_size, in SetupStreamsLocked()
206 input_stream_.get(), dataset()->options_.input_buffer_size, false); in SetupStreamsLocked()
213 input_stream_.reset(); in ResetStreamsLocked()
220 std::unique_ptr<io::RandomAccessInputStream> input_stream_
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsnapshot_util.cc661 input_stream_ = std::make_unique<io::RandomAccessInputStream>(file_.get()); in Initialize()
673 input_stream_ = absl::make_unique<io::ZlibInputStream>( in Initialize()
674 input_stream_.release(), zlib_options.input_buffer_size, in Initialize()
678 input_stream_ = absl::make_unique<io::SnappyInputBuffer>( in Initialize()
682 input_stream_ = in Initialize()
828 TF_RETURN_IF_ERROR(input_stream_->ReadNBytes(kHeaderSize, &header)); in ReadRecord()
830 return input_stream_->ReadNBytes(length, record); in ReadRecord()
836 TF_RETURN_IF_ERROR(input_stream_->ReadNBytes(kHeaderSize, &header)); in ReadRecord()
839 return input_stream_->ReadNBytes(length, record); in ReadRecord()
842 TF_RETURN_IF_ERROR(input_stream_->ReadNBytes(length, tmp_str)); in ReadRecord()
Dcsv_dataset_op.cc283 if (input_stream_) { in GetNextInternal()
331 if (input_stream_ && num_buffer_reads_ > 0) { in SaveInternal()
361 Status s = input_stream_->SkipNBytes( in RestoreInternal()
687 Status s = input_stream_->ReadNBytes( in FillBuffer()
858 input_stream_ = std::make_shared<io::ZlibInputStream>( in SetupStreamsLocked()
863 input_stream_ = random_access_input_stream_; in SetupStreamsLocked()
885 input_stream_.reset(); in ResetStreamsLocked()
896 std::shared_ptr<io::InputStreamInterface> input_stream_
Dsnapshot_util.h274 std::unique_ptr<io::InputStreamInterface> input_stream_; variable