Searched refs:AsyncWriter (Results 1 – 4 of 4) sorted by relevance
938 AsyncWriter::AsyncWriter(Env* env, int64 file_index, in AsyncWriter() function in tensorflow::data::snapshot_util::AsyncWriter952 void AsyncWriter::Write(const std::vector<Tensor>& tensors) { in Write()959 void AsyncWriter::SignalEOF() { in SignalEOF()966 void AsyncWriter::Consume(ElementOrEOF* be) { in Consume()968 mu_.Await(tensorflow::Condition(this, &AsyncWriter::ElementAvailable)); in Consume()973 bool AsyncWriter::ElementAvailable() { return !deque_.empty(); } in ElementAvailable()975 Status AsyncWriter::WriterThread(Env* env, const std::string& shard_directory, in WriterThread()
320 class AsyncWriter {322 explicit AsyncWriter(Env* env, int64 file_index,
96 absl::flat_hash_map<int64, std::unique_ptr<snapshot_util::AsyncWriter>> in WriteData()119 auto writer_thread = std::make_unique<snapshot_util::AsyncWriter>( in WriteData()
269 absl::flat_hash_map<int64, std::unique_ptr<snapshot_util::AsyncWriter>>704 snapshot_util::AsyncWriter* current_writer; in GetNextInternal()755 auto writer = std::make_unique<snapshot_util::AsyncWriter>( in GetNextInternal()