Home
last modified time | relevance | path

Searched refs:ReadNBytes (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/core/lib/io/
Drandom_inputstream_test.cc26 TEST(RandomInputStream, ReadNBytes) { in TEST() argument
35 TF_ASSERT_OK(in.ReadNBytes(3, &read)); in TEST()
38 TF_ASSERT_OK(in.ReadNBytes(0, &read)); in TEST()
41 TF_ASSERT_OK(in.ReadNBytes(5, &read)); in TEST()
44 TF_ASSERT_OK(in.ReadNBytes(0, &read)); in TEST()
47 EXPECT_TRUE(errors::IsOutOfRange(in.ReadNBytes(20, &read))); in TEST()
50 TF_ASSERT_OK(in.ReadNBytes(0, &read)); in TEST()
67 TF_ASSERT_OK(in.ReadNBytes(3, &read)); in TEST()
70 TF_ASSERT_OK(in.ReadNBytes(0, &read)); in TEST()
73 TF_ASSERT_OK(in.ReadNBytes(5, &read)); in TEST()
[all …]
Dinputbuffer_test.cc156 TEST(InputBuffer, ReadNBytes) { in TEST() argument
169 TF_CHECK_OK(in.ReadNBytes(3, &read)); in TEST()
172 TF_CHECK_OK(in.ReadNBytes(0, &read)); in TEST()
175 TF_CHECK_OK(in.ReadNBytes(4, &read)); in TEST()
178 TF_CHECK_OK(in.ReadNBytes(0, &read)); in TEST()
181 EXPECT_TRUE(errors::IsOutOfRange(in.ReadNBytes(5, &read))); in TEST()
184 EXPECT_TRUE(errors::IsOutOfRange(in.ReadNBytes(5, &read))); in TEST()
187 TF_CHECK_OK(in.ReadNBytes(0, &read)); in TEST()
200 TF_ASSERT_OK(in.ReadNBytes(3, read, &bytes_read)); in TEST()
204 TF_ASSERT_OK(in.ReadNBytes(0, read, &bytes_read)); in TEST()
[all …]
Dbuffered_inputstream_test.cc39 virtual Status ReadNBytes(int64 bytes_to_read, tstring* result) { in ReadNBytes() function in tensorflow::io::__anon59ed48cb0111::ReadOnceInputStream
192 TEST(BufferedInputStream, ReadNBytes) { in TEST() argument
206 TF_ASSERT_OK(in.ReadNBytes(3, &read)); in TEST()
209 TF_ASSERT_OK(in.ReadNBytes(0, &read)); in TEST()
212 TF_ASSERT_OK(in.ReadNBytes(4, &read)); in TEST()
215 TF_ASSERT_OK(in.ReadNBytes(0, &read)); in TEST()
218 EXPECT_TRUE(errors::IsOutOfRange(in.ReadNBytes(5, &read))); in TEST()
221 EXPECT_TRUE(errors::IsOutOfRange(in.ReadNBytes(5, &read))); in TEST()
224 TF_ASSERT_OK(in.ReadNBytes(0, &read)); in TEST()
239 TF_ASSERT_OK(in.ReadNBytes(3, &read)); in TEST()
[all …]
Dinputstream_interface_test.cc30 Status ReadNBytes(int64 bytes_to_read, tstring* result) override { in ReadNBytes() function in tensorflow::io::__anon12b2e9390111::TestStringStream
55 TF_ASSERT_OK(ss.ReadNBytes(4, &res)); in TEST()
58 TF_ASSERT_OK(ss.ReadNBytes(11, &res)); in TEST()
64 TF_ASSERT_OK(ss.ReadNBytes(4, &res)); in TEST()
Dinputstream_interface.h38 virtual Status ReadNBytes(int64 bytes_to_read, tstring* result) = 0;
44 virtual Status ReadNBytes(int64 bytes_to_read, absl::Cord* cord) { in ReadNBytes() function
Dzlib_buffers_test.cc92 TF_ASSERT_OK(in.ReadNBytes(data.size(), &result)); in TestAllCombinations()
150 TF_ASSERT_OK(in.ReadNBytes(data.size(), &decompressed_output)); in TestMultipleWrites()
195 Status read_status = in.ReadNBytes(data.size(), &result); in TEST()
243 TF_ASSERT_OK(in.ReadNBytes(first_half.size(), &bytes_read)); in TestTell()
251 in.ReadNBytes(data.size() - first_half.size(), &second_half)); in TestTell()
294 TF_ASSERT_OK(in.ReadNBytes(second_half.size(), &bytes_read)); in TestSkipNBytes()
324 EXPECT_TRUE(errors::IsDataLoss(in.ReadNBytes(5, &unused))); in TestSoftErrorOnDecompress()
Drandom_inputstream.h36 Status ReadNBytes(int64 bytes_to_read, tstring* result) override;
39 Status ReadNBytes(int64 bytes_to_read, absl::Cord* result) override;
Dzlib_inputstream.cc146 Status s = input_stream_->ReadNBytes(bytes_to_read, &data); in ReadFromStream()
196 Status ZlibInputStream::ReadNBytes(int64 bytes_to_read, tstring* result) { in ReadNBytes() function in tensorflow::io::ZlibInputStream
232 Status ZlibInputStream::ReadNBytes(int64 bytes_to_read, absl::Cord* result) { in ReadNBytes() function in tensorflow::io::ZlibInputStream
235 TF_RETURN_IF_ERROR(ReadNBytes(bytes_to_read, &buf)); in ReadNBytes()
Dzlib_inputstream.h69 Status ReadNBytes(int64 bytes_to_read, tstring* result) override;
72 Status ReadNBytes(int64 bytes_to_read, absl::Cord* result) override;
Dinputbuffer.cc79 Status InputBuffer::ReadNBytes(int64 bytes_to_read, string* result) { in ReadNBytes() function in tensorflow::io::InputBuffer
87 Status status = ReadNBytes(bytes_to_read, &(*result)[0], &bytes_read); in ReadNBytes()
92 Status InputBuffer::ReadNBytes(int64 bytes_to_read, char* result, in ReadNBytes() function in tensorflow::io::InputBuffer
148 TF_RETURN_IF_ERROR(ReadNBytes(1, p, &unused_bytes_read)); in ReadVarintFallback()
Dinputbuffer.h55 Status ReadNBytes(int64 bytes_to_read, string* result);
60 Status ReadNBytes(int64 bytes_to_read, char* result, size_t* bytes_read);
Drandom_inputstream.cc32 Status RandomAccessInputStream::ReadNBytes(int64 bytes_to_read, in ReadNBytes() function in tensorflow::io::RandomAccessInputStream
53 Status RandomAccessInputStream::ReadNBytes(int64 bytes_to_read, in ReadNBytes() function in tensorflow::io::RandomAccessInputStream
Dbuffered_inputstream.cc49 Status s = input_stream_->ReadNBytes(size_, &buf_); in FillBuffer()
89 Status BufferedInputStream::ReadNBytes(int64 bytes_to_read, tstring* result) { in ReadNBytes() function in tensorflow::io::BufferedInputStream
Dinputstream_interface.cc35 TF_RETURN_IF_ERROR(ReadNBytes(bytes_to_read, &unused)); in SkipNBytes()
Dbuffered_inputstream.h44 tensorflow::Status ReadNBytes(int64 bytes_to_read, tstring* result) override;
/external/tensorflow/tensorflow/core/lib/io/snappy/
Dsnappy_inputstream.cc46 Status SnappyInputStream::ReadNBytes(int64 bytes_to_read, tstring* result) { in ReadNBytes() function in tensorflow::io::SnappyInputStream
72 Status SnappyInputStream::ReadNBytes(int64 bytes_to_read, absl::Cord* result) { in ReadNBytes() function in tensorflow::io::SnappyInputStream
75 TF_RETURN_IF_ERROR(ReadNBytes(bytes_to_read, &buf)); in ReadNBytes()
87 input_stream_->ReadNBytes(sizeof(uint32), &compressed_block_length_ts)); in Inflate()
98 input_stream_->ReadNBytes(compressed_block_length, &compressed_block); in Inflate()
Dsnappy_inputstream.h47 Status ReadNBytes(int64 bytes_to_read, tstring* result) override;
50 Status ReadNBytes(int64 bytes_to_read, absl::Cord* result) override;
Dsnappy_test.cc148 TF_RETURN_IF_ERROR(in.ReadNBytes(data.size(), &decompressed_output)); in TestMultipleWrites()
187 snappy_input_stream.ReadNBytes(data.size(), &decompressed_output)); in TestMultipleWritesInputStream()
239 TF_CHECK_OK(in.ReadNBytes(first_half.size(), &bytes_read)); in TestTell()
246 TF_CHECK_OK(in.ReadNBytes(data.size() - first_half.size(), &second_half)); in TestTell()
276 TF_CHECK_OK(in.ReadNBytes(first_half.size(), &bytes_read)); in TestTellInputStream()
283 TF_CHECK_OK(in.ReadNBytes(data.size() - first_half.size(), &second_half)); in TestTellInputStream()
Dsnappy_inputbuffer.h57 Status ReadNBytes(int64 bytes_to_read, tstring* result) override;
Dsnappy_inputbuffer.cc33 Status SnappyInputBuffer::ReadNBytes(int64 bytes_to_read, tstring* result) { in ReadNBytes() function in tensorflow::io::SnappyInputBuffer
/external/tensorflow/tensorflow/core/kernels/
Ddecode_compressed_op.cc37 Status ReadNBytes(int64 bytes_to_read, tstring* result) override { in ReadNBytes() function in tensorflow::__anon384e59da0111::MemoryInputStream
110 Status s = zlib_stream->ReadNBytes(INT_MAX, &output_string); in Compute()
Dfixed_length_record_reader_op.cc108 Status s = buffered_inputstream_->ReadNBytes(bytes_to_read, value); in ReadLocked()
/external/tensorflow/tensorflow/core/kernels/data/
Dfixed_length_record_dataset_op.cc144 input_buffer_->ReadNBytes(dataset()->record_bytes_, &record)); in GetNextInternal()
270 TF_RETURN_IF_ERROR(buffered_input_stream_->ReadNBytes( in GetNextInternal()
283 Status s = buffered_input_stream_->ReadNBytes( in GetNextInternal()
371 TF_RETURN_IF_ERROR(buffered_input_stream_->ReadNBytes( in GetNextInternal()
428 TF_RETURN_IF_ERROR(buffered_input_stream_->ReadNBytes( in RestoreInternal()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsnapshot_util.cc828 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()
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc114 TF_RETURN_IF_ERROR(buffered_file->ReadNBytes( in ReadStringTensor()
137 buffered_file->ReadNBytes(string_length, &(*buffer)[0], &bytes_read)); in ReadStringTensor()
169 buffered_file->ReadNBytes(string_length, &buffer[0], &bytes_read)); in ReadVariantTensor()
188 TF_RETURN_IF_ERROR(buffered_file->ReadNBytes( in ReadVariantTensor()
898 TF_RETURN_IF_ERROR(buffered_file->ReadNBytes(entry.size(), backing_buffer, in GetValue()

12