Home
last modified time | relevance | path

Searched refs:ReadRecord (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/lib/io/
Drecord_reader_writer_test.cc92 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in VerifyFlush()
97 CHECK_EQ(reader.ReadRecord(&offset, &record).code(), error::OUT_OF_RANGE); in VerifyFlush()
146 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in TEST()
148 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in TEST()
191 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in TEST()
262 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in TEST()
264 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in TEST()
300 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in TEST()
302 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in TEST()
Drecordio_test.cc153 Status s = reader_->ReadRecord(&readpos_, &record); in Read()
187 Status s = reader_->ReadRecord(&offset, &record); in CheckOffsetPastEndReturnsNoRecords()
270 TF_ASSERT_OK(reader.ReadRecord(&offset, &record)) << i; in TestNonSequentialReads()
275 TF_ASSERT_OK(reader.ReadRecord(&offset, &record)); in TestNonSequentialReads()
280 TF_ASSERT_OK(reader.ReadRecord(&offset, &record)); in TestNonSequentialReads()
320 Status status = reader.ReadRecord(&offset, &read); in TestReadError()
326 status = reader.ReadRecord(&offset, &read); in TestReadError()
Drecord_reader.h98 Status ReadRecord(uint64* offset, tstring* record);
146 Status ReadRecord(tstring* record) { in ReadRecord() function
147 return underlying_.ReadRecord(&offset_, record); in ReadRecord()
Drecord_reader.cc185 Status RecordReader::ReadRecord(uint64* offset, tstring* record) { in ReadRecord() function in tensorflow::io::RecordReader
/external/tensorflow/tensorflow/python/lib/io/
Drecord_io_wrapper.cc64 return reader_->ReadRecord(&offset_, out); in ReadNextRecord()
107 tensorflow::Status ReadRecord(tensorflow::uint64* offset, in ReadRecord() function in __anonab493b040111::PyRecordRandomReader
113 return reader_->ReadRecord(offset, out); in ReadRecord()
257 status = self->ReadRecord(&temp_offset, &record); in PYBIND11_MODULE()
Dpy_record_reader.cc67 Status s = reader_->ReadRecord(&offset_, &record_); in GetNext()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_resource_ops_test.cc200 TF_ASSERT_OK(reader->ReadRecord(&offset, &record)); in TEST_F()
207 EXPECT_TRUE(errors::IsOutOfRange(reader->ReadRecord(&offset, &record))); in TEST_F()
Dtrt_engine_resource_ops.cc124 Status status = reader->ReadRecord(&offset, &record); in Compute()
/external/tensorflow/tensorflow/core/data/service/
Dutils.cc46 TF_RETURN_IF_ERROR(reader.ReadRecord(&offset, &record)); in ReadDatasetDef()
Djournal.cc103 Status s = reader_->ReadRecord(&offset_, &record); in Read()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsnapshot_util.cc636 TF_RETURN_IF_ERROR(record_reader_->ReadRecord(&offset_, &record)); in ReadTensors()
718 TF_RETURN_IF_ERROR(ReadRecord(&metadata_str)); in ReadTensors()
759 TF_RETURN_IF_ERROR(ReadRecord(&c)); in ReadTensorsV0()
763 TF_RETURN_IF_ERROR(ReadRecord(&record_bytes)); in ReadTensorsV0()
782 TF_RETURN_IF_ERROR(ReadRecord(&compressed)); in SnappyUncompress()
826 Status CustomReader::ReadRecord(tstring* record) { in ReadRecord() function in tensorflow::data::snapshot_util::CustomReader
834 Status CustomReader::ReadRecord(absl::Cord* record) { in ReadRecord() function in tensorflow::data::snapshot_util::CustomReader
Dsnapshot_util.h266 Status ReadRecord(tstring* record);
269 Status ReadRecord(absl::Cord* record);
Dcsv_dataset_op.cc285 ReadRecord(ctx, out_tensors, select_all, in GetNextInternal()
389 Status ReadRecord(IteratorContext* ctx, std::vector<Tensor>* out_tensors, in ReadRecord() function in tensorflow::data::experimental::__anon7230ae400111::CSVDatasetOp::Dataset::Iterator
875 Status s = ReadRecord(nullptr, nullptr, false, empty, empty); in SetupStreamsLocked()
/external/tensorflow/tensorflow/core/summary/
Dsummary_file_writer_test.cc75 reader.ReadRecord(&offset, in SummaryTestHelper()
77 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in SummaryTestHelper()
Dloader.cc102 Status s = reader.ReadRecord(&offset, &record); in main()
/external/tensorflow/tensorflow/core/kernels/
Dtf_record_reader_op.cc56 Status status = reader_->ReadRecord(&offset_, value); in ReadLocked()
Drecord_yielder.cc216 Status s = rdr.ReadRecord(&offset, &record); in ShardLoop()
/external/tensorflow/tensorflow/core/util/
Devents_writer_test.cc57 Status s = reader->ReadRecord(offset, &record); in ReadEventProto()
Ddebug_events_writer_test.cc63 Status s = reader->ReadRecord(offset, &record); in ReadDebugEventProto()
/external/tensorflow/tensorflow/core/kernels/data/
Dtf_record_dataset_op.cc126 reader_->ReadRecord(&out_tensors->back().scalar<tstring>()()); in GetNextInternal()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc364 while (reader.ReadRecord(&offset, &record).ok()) { in ParseRecordIoFile()