Fix DCHECK not defined issue in profiler.

PiperOrigin-RevId: 286256041 Change-Id: Ie61edb82e440deda698d21b7655ecde550c97ca2
2019-12-18 13:40:24 -08:00 · 2019-12-18 13:40:24 -08:00 · 84d5bf11e3
commit 84d5bf11e3
parent 201f19e439
2 changed files with 3 additions and 0 deletions
--- a/tensorflow/core/profiler/convert/BUILD
+++ b/tensorflow/core/profiler/convert/BUILD
@ -40,6 +40,7 @@ cc_library(
    srcs = ["op_metrics_db_combiner.cc"],
    hdrs = ["op_metrics_db_combiner.h"],
    deps = [
+        "//tensorflow/core/platform:logging",
        "//tensorflow/core/profiler/protobuf:op_metrics_proto_cc",
        "//tensorflow/core/profiler/utils:op_metrics_db_utils",
    ],
--- a/tensorflow/core/profiler/convert/op_metrics_db_combiner.cc
+++ b/tensorflow/core/profiler/convert/op_metrics_db_combiner.cc
@ -15,6 +15,8 @@ limitations under the License.

 #include "tensorflow/core/profiler/convert/op_metrics_db_combiner.h"

+#include "tensorflow/core/platform/logging.h"
+
 namespace tensorflow {
 namespace profiler {
 namespace {