Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Ddebug_events_writer.cc42 if (record_writer_ != nullptr) { in Init()
50 record_writer_.reset(); in Init()
55 record_writer_.reset(new io::RecordWriter(writable_file_.get())); in Init()
56 if (record_writer_ == nullptr) { in Init()
67 if (record_writer_ == nullptr) { in WriteSerializedDebugEvent()
76 record_writer_->WriteRecord(debug_event_str).IgnoreError(); in WriteSerializedDebugEvent()
91 TF_RETURN_WITH_CONTEXT_IF_ERROR(record_writer_->Flush(), "Failed to flush ", in Flush()
110 record_writer_.reset(nullptr); in Close()
Ddebug_events_writer.h66 std::unique_ptr<io::RecordWriter> record_writer_ TF_PT_GUARDED_BY(writer_mu_);
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsnapshot_util.cc110 record_writer_ = absl::make_unique<io::RecordWriter>( in Initialize()
129 TF_RETURN_IF_ERROR(record_writer_->WriteRecord(proto_serialized)); in WriteTensors()
131 TF_RETURN_IF_ERROR(record_writer_->WriteRecord(proto.SerializeAsString())); in WriteTensors()
138 TF_RETURN_IF_ERROR(record_writer_->Flush()); in Sync()
143 if (record_writer_ != nullptr) { in Close()
145 TF_RETURN_IF_ERROR(record_writer_->Close()); in Close()
147 record_writer_ = nullptr; in Close()
Dsnapshot_util.h119 std::unique_ptr<io::RecordWriter> record_writer_; variable