experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add TF_LOCKS_EXCLUDED to MutableDenseHashTable::MemoryUsed

Browse Source
This commit is contained in:
rangjiaheng 2020-05-27 13:27:18 +08:00
parent 8f31b06f53
commit b4062c71c6
1 changed files with 1 additions and 1 deletions

2
tensorflow/core/kernels/lookup_table_op.cc
View File

@ -557,7 +557,7 @@ class MutableDenseHashTable final : public LookupInterface {
TensorShape value_shape() const override { return value_shape_; } 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_); tf_shared_lock l(mu_);
return sizeof(MutableDenseHashTable) + key_buckets_.AllocatedBytes() + return sizeof(MutableDenseHashTable) + key_buckets_.AllocatedBytes() +
value_buckets_.AllocatedBytes() + empty_key_.AllocatedBytes(); value_buckets_.AllocatedBytes() + empty_key_.AllocatedBytes();