Fix python_hooks BUILD deps

PiperOrigin-RevId: 302727575
Change-Id: I644dfa679cd92aa5e24afb6885943057d324d048

tensorflow/python/profiler/internal/BUILD

@@ -134,11 +134,12 @@ cc_library(
     srcs = ["python_hooks.cc"],
     hdrs = ["python_hooks.h"],
     copts = ["-fexceptions"],
-    features = ["-use_header_modules"],
+    features = ["-use_header_modules"],  # Incompatible with -fexceptions.
     visibility = ["//visibility:public"],
     deps = [
-        "//tensorflow/core/platform:path",
+        "//tensorflow/core:lib",
         "//tensorflow/core/profiler/lib:traceme",
+        "@com_google_absl//absl/container:flat_hash_map",
         "@com_google_absl//absl/strings",
         "@pybind11",
     ],

tensorflow/python/profiler/internal/python_hooks.cc

@@ -21,7 +21,6 @@ namespace tensorflow {
 namespace profiler {
 
 namespace py = ::pybind11;
-using tensorflow::profiler::TraceMe;
 
 template <typename T>
 int ProfileFunction(PyObject* obj, PyFrameObject* frame, int what,

tensorflow/python/profiler/internal/python_hooks.h

@@ -15,6 +15,9 @@ limitations under the License.
 #ifndef TENSORFLOW_PYTHON_PROFILER_INTERNAL_PYTHON_HOOKS_H_
 #define TENSORFLOW_PYTHON_PROFILER_INTERNAL_PYTHON_HOOKS_H_
 
+#include <memory>
+#include <vector>
+
 #include "absl/container/flat_hash_map.h"
 #include "include/pybind11/cast.h"
 #include "include/pybind11/pybind11.h"