Searched refs:CustomWriter (Results 1 – 4 of 4) sorted by relevance
89 absl::make_unique<CustomWriter>(filename, compression_type, dtypes); in Create()160 CustomWriter::CustomWriter(const std::string& filename, in CustomWriter() function in tensorflow::data::snapshot_util::CustomWriter167 Status CustomWriter::Initialize(tensorflow::Env* env) { in Initialize()201 Status CustomWriter::WriteTensors(const std::vector<Tensor>& tensors) { in WriteTensors()284 Status CustomWriter::Sync() { return dest_->Sync(); } in Sync()286 Status CustomWriter::Close() { in Close()298 CustomWriter::~CustomWriter() { in ~CustomWriter()305 Status CustomWriter::WriteRecord(const StringPiece& data) { in WriteRecord()313 Status CustomWriter::WriteRecord(const absl::Cord& data) { in WriteRecord()
123 class CustomWriter : public Writer {132 CustomWriter(const std::string& filename, const std::string& compression_type,141 ~CustomWriter() override;
81 typedef flatbuffers::Offset<Operator> (*CustomWriter)( typedef106 CustomWriter custom_writer);198 std::unordered_map<std::string, CustomWriter> custom_op_to_writer_;
319 const std::string& custom_name, CustomWriter custom_writer) { in RegisterCustomWriter()