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 #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"