Automated Code Change

PiperOrigin-RevId: 625280358
tensorflow · Apr 16, 2024 · dd167e3 · dd167e3
1 parent fc84fe9
commit dd167e3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tensorflow/python/profiler/internal/profiler_pywrap_impl.cc b/tensorflow/python/profiler/internal/profiler_pywrap_impl.cc
@@ -74,7 +74,7 @@ tensorflow::Status ProfilerSessionWrapper::Start(
     const absl::flat_hash_map<std::string, std::variant<int, std::string>>&
         options) {
   auto opts = GetRemoteSessionManagerOptionsLocked(logdir, options);
-  session_ = tensorflow::ProfilerSession::Create(opts.profiler_options());
+  session_ = tsl::ProfilerSession::Create(opts.profiler_options());
   logdir_ = logdir;
   return session_->Status();
 }

diff --git a/tensorflow/python/profiler/internal/profiler_pywrap_impl.h b/tensorflow/python/profiler/internal/profiler_pywrap_impl.h
@@ -47,7 +47,7 @@ class ProfilerSessionWrapper {
   tensorflow::Status ExportToTensorBoard();
 
  private:
-  std::unique_ptr<tensorflow::ProfilerSession> session_;
+  std::unique_ptr<tsl::ProfilerSession> session_;
   tensorflow::string logdir_;
 };