Searched refs:size_map_ (Results 1 – 1 of 1) sorted by relevance
32 size_map_[ptr] = num_bytes; in AllocateRaw()36 const auto& iter = size_map_.find(ptr); in DeallocateRaw()37 EXPECT_NE(size_map_.end(), iter); in DeallocateRaw()38 size_map_.erase(iter); in DeallocateRaw()43 const auto& iter = size_map_.find(ptr); in RequestedSize()44 EXPECT_NE(size_map_.end(), iter); in RequestedSize()50 std::unordered_map<const void*, size_t> size_map_; member in tensorflow::TestableSizeTrackingAllocator