Automated Code Change

PiperOrigin-RevId: 849601638

Author: tensorflower-gardener

tensorflow/core/tpu/kernels/tpu_compilation_cache_entry_unloader.h

@@ -40,7 +40,7 @@ class TpuCompilationCacheEntryUnloader : public ResourceBase {
   }
 
   ~TpuCompilationCacheEntryUnloader() override {
-    absl::MutexLock lock(&mu_);
+    absl::MutexLock lock(mu_);
     for (int64_t uid : cache_entry_uids_) {
       absl::Status s = cache_->MarkEntryForEviction(uid);
       if (!s.ok()) {
@@ -55,7 +55,7 @@ class TpuCompilationCacheEntryUnloader : public ResourceBase {
 
   // Add cache entry uid to be unloaded in destructor.
   void AddCacheEntryUid(int64_t uid) {
-    absl::MutexLock lock(&mu_);
+    absl::MutexLock lock(mu_);
     cache_entry_uids_.insert(uid);
   }
 

tensorflow/core/tpu/kernels/tpu_compilation_cache_interface.cc

@@ -153,7 +153,7 @@ absl::Status TpuCompilationCacheInterface::MarkEntryForEviction(
       /*level=*/2);
   CompiledSubgraph* deleted_entry = nullptr;
   {
-    absl::MutexLock lock(&mu_);
+    absl::MutexLock lock(mu_);
     auto iter = entries_by_uid_.find(subgraph_uid);
     if (iter == entries_by_uid_.end()) {
       // If already evicted, return ok.
@@ -201,7 +201,7 @@ absl::Status TpuCompilationCacheInterface::Release(int64_t subgraph_uid) {
 
   CompiledSubgraph* deleted_entry = nullptr;
   {
-    absl::MutexLock lock(&mu_);
+    absl::MutexLock lock(mu_);
     auto iter = entries_by_uid_.find(subgraph_uid);
 
     if (iter == entries_by_uid_.end()) {
@@ -291,7 +291,7 @@ void TpuCompilationCacheInterface::DiscardEntryRefs(
     absl::Span<CompiledSubgraph* const> entries) {
   std::vector<CompiledSubgraph*> removed_entries;
   {
-    absl::MutexLock lock(&mu_);
+    absl::MutexLock lock(mu_);
 
     for (auto entry : entries) {
       removed_entries.push_back(DiscardEntryRef(entry));
@@ -424,7 +424,7 @@ absl::Status TpuCompilationCacheInterface::CompileIfKeyAbsentHelper(
 
   // NOTE: In spite of the fact that we use MutexLock, we do not hold the lock
   // for the lifetime of the object, see InitializeEntry() call below.
-  absl::MutexLock lock(&mu_);
+  absl::MutexLock lock(mu_);
 
   std::string cache_key = FindCacheKey(subgraph_key);
   auto iter = cache_.find(cache_key);
@@ -556,7 +556,7 @@ absl::Status TpuCompilationCacheInterface::GetKeysFromUid(
     int64_t uid, std::vector<std::string>* keys) {
   keys->clear();
 
-  absl::MutexLock lock(&mu_);
+  absl::MutexLock lock(mu_);
   const auto iter = entries_by_uid_.find(uid);
   if (iter == entries_by_uid_.end()) {
     return errors::NotFound("No subgraph found for uid ", uid);
@@ -574,7 +574,7 @@ absl::Status TpuCompilationCacheInterface::Lookup(
       "TPU compilation cache proto lookup by uid",
       /*level=*/2);
 
-  absl::MutexLock lock(&mu_);
+  absl::MutexLock lock(mu_);
   const auto iter = entries_by_uid_.find(uid);
   if (iter == entries_by_uid_.end()) {
     return errors::NotFound("No subgraph found for uid ", uid);
@@ -599,7 +599,7 @@ absl::Status TpuCompilationCacheInterface::Lookup(
       "TPU compilation cache proto lookup",
       /*level=*/2);
 
-  absl::MutexLock lock(&mu_);
+  absl::MutexLock lock(mu_);
   const auto iter = entries_by_proto_key_.find(proto_key);
   if (iter == entries_by_proto_key_.end()) {
     return errors::NotFound("No proto found for key ", proto_key);

tensorflow/core/tpu/kernels/tpu_compilation_cache_rpc_lookup.cc

@@ -81,7 +81,7 @@ absl::Status TpuCompilationCacheRpcLookup::Lookup(
       proto_key, "_", tpu::CompilationCacheFetchTarget_Name(fetch_target));
 
   {
-    absl::MutexLock lock(&mu_);
+    absl::MutexLock lock(mu_);
     auto iter = cache_.find(local_proto_key);
     if (iter == cache_.end()) {
       tpu::GetTpuProgramRequest request;
@@ -123,7 +123,7 @@ absl::Status TpuCompilationCacheRpcLookup::Lookup(
       absl::StrCat(" _ ", uid, ":", proto_index, "_",
                    tpu::CompilationCacheFetchTarget_Name(fetch_target));
   {
-    absl::MutexLock lock(&mu_);
+    absl::MutexLock lock(mu_);
     auto iter = cache_.find(local_proto_key);
     if (iter == cache_.end()) {
       tpu::GetTpuProgramRequest request;

tensorflow/core/tpu/kernels/tpu_fingerprint_lookup.cc

@@ -26,7 +26,7 @@ TpuFingerprintLookup* TpuFingerprintLookup::Create() {
 
 void TpuFingerprintLookup::RegisterKeyAndIntermediatePair(uint64 key,
                                                           uint64 intermediate) {
-  absl::MutexLock lock(&mu_);
+  absl::MutexLock lock(mu_);
   auto [it, emplaced] = intermediate_to_key_.try_emplace(intermediate, key);
   if (it->second != key) {
     VLOG(2) << "The key (" << it->second
@@ -37,7 +37,7 @@ void TpuFingerprintLookup::RegisterKeyAndIntermediatePair(uint64 key,
 
 bool TpuFingerprintLookup::RegisterIntermediateAndValuePair(uint64 intermediate,
                                                             std::string value) {
-  absl::MutexLock lock(&mu_);
+  absl::MutexLock lock(mu_);
   auto it = intermediate_to_key_.find(intermediate);
   if (it == intermediate_to_key_.end()) {
     VLOG(2) << "Cannot find the intermediate ( " << intermediate
@@ -83,7 +83,7 @@ bool TpuFingerprintLookup::RegisterIntermediateAndValuePair(uint64 intermediate,
 }
 
 std::optional<absl::string_view> TpuFingerprintLookup::Lookup(uint64 key) {
-  absl::MutexLock lock(&mu_);
+  absl::MutexLock lock(mu_);
   auto it = key_to_value_.find(key);
   if (it == key_to_value_.end()) {
     return std::optional<absl::string_view>{};

tensorflow/core/tpu/kernels/tpu_fingerprint_lookup.h

@@ -68,7 +68,7 @@ class TpuFingerprintLookup : public ResourceBase {
   std::optional<absl::string_view> Lookup(uint64 key);
 
   size_t num_valid() {
-    absl::MutexLock lock(&mu_);
+    absl::MutexLock lock(mu_);
     return key_to_value_.size();
   }
 

tensorflow/core/tpu/kernels/tpu_functional_ops.cc

@@ -1263,7 +1263,7 @@ void TPUPartitionedCallOp::ComputeAsync(OpKernelContext* ctx,
     bool enable_spmd_xla_partitioning = false;
     TPUMetadata tpu_metadata;
     {
-      absl::MutexLock l(&mu_);
+      absl::MutexLock l(mu_);
       // We are not using OP_REQUIRES_OK_ASYNC here as we are inside the
       // call_once. It will be checked later whether `ordinal_selector_` is
       // initialized.
@@ -1301,7 +1301,7 @@ void TPUPartitionedCallOp::ComputeAsync(OpKernelContext* ctx,
                         &device_ordinal),
       done);
   uint64_t cache_hash = Hash64Combine(input_hash, device_ordinal);
-  absl::ReleasableMutexLock lock(&mu_);
+  absl::ReleasableMutexLock lock(mu_);
 
   const std::vector<DeviceAndFHandle>* functions;