diff options
Diffstat (limited to 'tensorflow/contrib/tensorboard/db/summary_db_writer.cc')
-rw-r--r-- | tensorflow/contrib/tensorboard/db/summary_db_writer.cc | 34 |
1 files changed, 4 insertions, 30 deletions
diff --git a/tensorflow/contrib/tensorboard/db/summary_db_writer.cc b/tensorflow/contrib/tensorboard/db/summary_db_writer.cc index a26ad61660..df64e36305 100644 --- a/tensorflow/contrib/tensorboard/db/summary_db_writer.cc +++ b/tensorflow/contrib/tensorboard/db/summary_db_writer.cc @@ -15,12 +15,10 @@ limitations under the License. #include "tensorflow/contrib/tensorboard/db/summary_db_writer.h" #include "tensorflow/contrib/tensorboard/db/schema.h" -#include "tensorflow/core/framework/summary.pb.h" #include "tensorflow/core/lib/db/sqlite.h" #include "tensorflow/core/lib/random/random.h" #include "tensorflow/core/lib/strings/stringprintf.h" #include "tensorflow/core/platform/snappy.h" -#include "tensorflow/core/util/event.pb.h" namespace tensorflow { namespace { @@ -88,19 +86,13 @@ class SummaryDbWriter : public SummaryWriterInterface { TF_RETURN_IF_ERROR(BindTensor(t)); break; } - return insert_tensor_.StepAndReset(); + TF_RETURN_IF_ERROR(insert_tensor_.StepAndReset()); + return Status::OK(); } Status WriteEvent(std::unique_ptr<Event> e) override { - mutex_lock ml(mu_); - TF_RETURN_IF_ERROR(InitializeParents()); - if (e->what_case() == Event::WhatCase::kSummary) { - const Summary& summary = e->summary(); - for (int i = 0; i < summary.value_size(); ++i) { - TF_RETURN_IF_ERROR(WriteSummary(e.get(), summary.value(i))); - } - } - return Status::OK(); + // TODO(@jart): This will be used to load event logs. + return errors::Unimplemented("WriteEvent"); } Status WriteScalar(int64 global_step, Tensor t, const string& tag) override { @@ -255,24 +247,6 @@ class SummaryDbWriter : public SummaryWriterInterface { return Status::OK(); } - Status WriteSummary(const Event* e, const Summary::Value& summary) - EXCLUSIVE_LOCKS_REQUIRED(mu_) { - int64 tag_id; - TF_RETURN_IF_ERROR(GetTagId(run_id_, summary.tag(), &tag_id)); - insert_tensor_.BindInt(1, tag_id); - insert_tensor_.BindInt(2, e->step()); - insert_tensor_.BindDouble(3, e->wall_time()); - switch (summary.value_case()) { - case Summary::Value::ValueCase::kSimpleValue: - insert_tensor_.BindDouble(4, summary.simple_value()); - break; - default: - // TODO(@jart): Handle the rest. - return Status::OK(); - } - return insert_tensor_.StepAndReset(); - } - mutex mu_; Env* env_; std::shared_ptr<Sqlite> db_ GUARDED_BY(mu_); |