aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/tensorboard/db/summary_db_writer.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/tensorboard/db/summary_db_writer.cc')
-rw-r--r--tensorflow/contrib/tensorboard/db/summary_db_writer.cc34
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_);