Searched refs:WriteTensors (Results 1 – 4 of 4) sorted by relevance
55 TF_ASSERT_OK(writer->WriteTensors(tensors)); in SnapshotRoundTrip()108 writer->WriteTensors(tensors).IgnoreError(); in SnapshotReaderBenchmarkLoop()164 writer->WriteTensors(tensors).IgnoreError(); in SnapshotWriterBenchmarkLoop()
82 virtual Status WriteTensors(const std::vector<Tensor>& tensors) = 0;103 Status WriteTensors(const std::vector<Tensor>& tensors) override;135 Status WriteTensors(const std::vector<Tensor>& tensors) override;
116 Status TFRecordWriter::WriteTensors(const std::vector<Tensor>& tensors) { in WriteTensors() function in tensorflow::data::snapshot_util::TFRecordWriter201 Status CustomWriter::WriteTensors(const std::vector<Tensor>& tensors) { in WriteTensors() function in tensorflow::data::snapshot_util::CustomWriter995 TF_RETURN_IF_ERROR(writer->WriteTensors(be.value)); in WriterThread()
2169 TF_RETURN_IF_ERROR((*writer)->WriteTensors(elem.value)); in ProcessOneElement()