Searched refs:allocation_map (Results 1 – 1 of 1) sorted by relevance
220 AllocationMap& allocation_map = opaque_to_allocation_map_[device_ordinal]; in AddAllocationOrIncrementRefCount() local221 auto it = allocation_map.find(device_memory.opaque()); in AddAllocationOrIncrementRefCount()222 if (it == allocation_map.end()) { in AddAllocationOrIncrementRefCount()223 allocation_map[device_memory.opaque()] = { in AddAllocationOrIncrementRefCount()234 AllocationMap& allocation_map = opaque_to_allocation_map_[device_ordinal]; in DecrementRefCount() local235 auto it = allocation_map.find(device_memory.opaque()); in DecrementRefCount()236 TF_RET_CHECK(it != allocation_map.end()); in DecrementRefCount()241 allocation_map.erase(it); in DecrementRefCount()