Searched refs:read_tensors (Results 1 – 4 of 4) sorted by relevance
64 std::vector<Tensor> read_tensors; in SnapshotRoundTrip() local65 TF_ASSERT_OK(reader->ReadTensors(&read_tensors)); in SnapshotRoundTrip()66 EXPECT_EQ(tensors.size(), read_tensors.size()); in SnapshotRoundTrip()67 for (int j = 0; j < read_tensors.size(); ++j) { in SnapshotRoundTrip()72 read_tensors[j].AsProtoTensorContent(&read_proto); in SnapshotRoundTrip()117 std::vector<Tensor> read_tensors; in SnapshotReaderBenchmarkLoop() local118 reader->ReadTensors(&read_tensors).IgnoreError(); in SnapshotReaderBenchmarkLoop()
632 Status TFRecordReader::ReadTensors(std::vector<Tensor>* read_tensors) { in ReadTensors() argument633 read_tensors->reserve(dtypes_.size()); in ReadTensors()646 read_tensors->push_back(std::move(tensor)); in ReadTensors()701 Status CustomReader::ReadTensors(std::vector<Tensor>* read_tensors) { in ReadTensors() argument706 return ReadTensorsV0(read_tensors); in ReadTensors()722 read_tensors->reserve(metadata.tensor_metadata_size()); in ReadTensors()735 read_tensors->push_back(std::move(simple_tensors[simple_index])); in ReadTensors()748 read_tensors->push_back(std::move(t)); in ReadTensors()755 Status CustomReader::ReadTensorsV0(std::vector<Tensor>* read_tensors) { in ReadTensorsV0() argument766 read_tensors->reserve(record.tensor_size()); in ReadTensorsV0()[all …]
191 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;258 Status ReadTensorsV0(std::vector<Tensor>* read_tensors);
1605 std::vector<Tensor> read_tensors; in ReadFile() local1606 Status s = reader->ReadTensors(&read_tensors); in ReadFile()1612 elem.value = std::move(read_tensors); in ReadFile()