Automated Code Change

PiperOrigin-RevId: 623116061
tensorflow · Apr 9, 2024 · 050de33 · 050de33
1 parent 8d2267d
commit 050de33
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tensorflow/core/profiler/lib/profiler_disabled_test.cc b/tensorflow/core/profiler/lib/profiler_disabled_test.cc
@@ -24,7 +24,7 @@ namespace {
 
 TEST(ProfilerDisabledTest, ProfilerDisabledTest) {
   setenv("TF_DISABLE_PROFILING", "1", /*overwrite=*/1);
-  StatusOr<ProfilerLock> profiler_lock = ProfilerLock::Acquire();
+  StatusOr<ProfilerLock> profiler_lock = tsl::profiler::ProfilerLock::Acquire();
   EXPECT_FALSE(profiler_lock.ok());
 }
 

diff --git a/tensorflow/core/profiler/lib/profiler_factory.h b/tensorflow/core/profiler/lib/profiler_factory.h
@@ -40,7 +40,7 @@ using ProfilerFactor ABSL_DEPRECATE_AND_INLINE() =
 
 // Registers a profiler factory. Should be invoked at most once per factory.
 ABSL_DEPRECATE_AND_INLINE()
-inline void RegisterProfilerFactory(ProfilerFactor factory) {
+inline void RegisterProfilerFactory(tsl::profiler::ProfilerFactory factory) {
   tsl::profiler::RegisterProfilerFactory(std::move(factory));
 }