Home
last modified time | relevance | path

Searched refs:ReadTensors (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsnapshot_util.h191 virtual Status ReadTensors(std::vector<Tensor>* read_tensors) = 0;
212 Status ReadTensors(std::vector<Tensor>* read_tensors) override;
250 Status ReadTensors(std::vector<Tensor>* read_tensors) override;
Dsnapshot_util_test.cc65 TF_ASSERT_OK(reader->ReadTensors(&read_tensors)); in SnapshotRoundTrip()
118 reader->ReadTensors(&read_tensors).IgnoreError(); in SnapshotReaderBenchmarkLoop()
Dsnapshot_util.cc350 TF_RETURN_IF_ERROR(ReadTensors(&unused_tensors)); in SkipRecords()
424 Status s = reader_->ReadTensors(out_tensors); in GetNextInternal()
632 Status TFRecordReader::ReadTensors(std::vector<Tensor>* read_tensors) { in ReadTensors() function in tensorflow::data::snapshot_util::TFRecordReader
701 Status CustomReader::ReadTensors(std::vector<Tensor>* read_tensors) { in ReadTensors() function in tensorflow::data::snapshot_util::CustomReader
Dsnapshot_dataset_op.cc1606 Status s = reader->ReadTensors(&read_tensors); in ReadFile()