diff options
| author |  Yuefeng Zhou <yuefengz@google.com> | 2017-03-02 12:57:39 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-02 13:10:14 -0800 |
| commit | 02e53cf8e37e0f0228520167dbb1e12632f6dd48 (patch) | |
| tree | d53917ef38284bb53782334d3a75cc7a1f120f53 /tensorflow/core/util/presized_cuckoo_map.h | |
| parent | 87ce73a983a01da6e39530edad2d242046644243 (diff) | |
Tracking lookup table memory usage.
Change: 149036604
Diffstat (limited to 'tensorflow/core/util/presized_cuckoo_map.h')
| -rw-r--r-- | tensorflow/core/util/presized_cuckoo_map.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/util/presized_cuckoo_map.h b/tensorflow/core/util/presized_cuckoo_map.h index cf3b8cf5b3..e7dab830f0 100644 --- a/tensorflow/core/util/presized_cuckoo_map.h +++ b/tensorflow/core/util/presized_cuckoo_map.h @@ -132,6 +132,10 @@ class PresizedCuckooMap { FindInBucket(k, fast_map_to_buckets(h2(tk)), out); } + int64 MemoryUsed() const { + return sizeof(PresizedCuckooMap<value>) + sizeof(CuckooPathQueue); + } + private: static constexpr int kSlotsPerBucket = 4; |