From 8b5e015c512f14c0bb3ade1054f0d010d2d5af24 Mon Sep 17 00:00:00 2001
From: Adrian Kuegel <akuegel@google.com>
Date: Tue, 6 Oct 2020 09:27:35 -0700
Subject: [PATCH] Fix build by adding missing include.

PiperOrigin-RevId: 335655957
Change-Id: I1c6b1c03e255c1f347df940db2eca0a516243977
---
 tensorflow/core/profiler/rpc/client/BUILD                       | 2 ++
 tensorflow/core/profiler/rpc/client/profiler_client_test_util.h | 1 +
 2 files changed, 3 insertions(+)

diff --git a/tensorflow/core/profiler/rpc/client/BUILD b/tensorflow/core/profiler/rpc/client/BUILD
index f91dde7bb77..2c552a9864c 100644
--- a/tensorflow/core/profiler/rpc/client/BUILD
+++ b/tensorflow/core/profiler/rpc/client/BUILD
@@ -85,6 +85,7 @@ tf_cc_test(
         ":profiler_client_impl",
         "@com_google_absl//absl/memory",
         "@com_google_absl//absl/time",
+        "@com_google_absl//absl/strings:str_format",
         "//tensorflow/core:test",
         "//tensorflow/core:test_main",
         "//tensorflow/core/lib/core:errors",
@@ -123,6 +124,7 @@ tf_cc_test(
     tags = ["external"],  # So that test suite reruns unconditionally.
     deps = [
         ":remote_profiler_session_manager",
+        "@com_google_absl//absl/strings:str_format",
         "@com_google_absl//absl/memory",
         "@com_google_absl//absl/time",
         "//tensorflow/core:test",
diff --git a/tensorflow/core/profiler/rpc/client/profiler_client_test_util.h b/tensorflow/core/profiler/rpc/client/profiler_client_test_util.h
index 925f174f947..79afbe02bf5 100644
--- a/tensorflow/core/profiler/rpc/client/profiler_client_test_util.h
+++ b/tensorflow/core/profiler/rpc/client/profiler_client_test_util.h
@@ -21,6 +21,7 @@ limitations under the License.
 #include <string>
 
 #include "absl/memory/memory.h"
+#include "absl/strings/str_format.h"
 #include "absl/time/clock.h"
 #include "absl/time/time.h"
 #include "tensorflow/core/platform/platform.h"