From b4062c71c659bde06e5e15c53424a6853aa1ee3c Mon Sep 17 00:00:00 2001
From: rangjiaheng <rangjiaheng@gmail.com>
Date: Wed, 27 May 2020 13:27:18 +0800
Subject: [PATCH] add TF_LOCKS_EXCLUDED to MutableDenseHashTable::MemoryUsed
---
tensorflow/core/kernels/lookup_table_op.cc | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tensorflow/core/kernels/lookup_table_op.cc b/tensorflow/core/kernels/lookup_table_op.cc
index 9807247ed4f..f269aa65b4e 100644
--- a/tensorflow/core/kernels/lookup_table_op.cc
+++ b/tensorflow/core/kernels/lookup_table_op.cc
@@ -557,7 +557,7 @@ class MutableDenseHashTable final : public LookupInterface {
TensorShape value_shape() const override { return value_shape_; }
- int64 MemoryUsed() const override {
+ int64 MemoryUsed() const override TF_LOCKS_EXCLUDED(mu_) {
tf_shared_lock l(mu_);
return sizeof(MutableDenseHashTable) + key_buckets_.AllocatedBytes() +
value_buckets_.AllocatedBytes() + empty_key_.AllocatedBytes();