Searched refs:allocation_map (Results 1 – 1 of 1) sorted by relevance
208 AllocationMap& allocation_map = opaque_to_allocation_map_[device_ordinal]; in AddAllocationOrIncrementRefCount() local209 auto it = allocation_map.find(device_memory.opaque()); in AddAllocationOrIncrementRefCount()210 if (it == allocation_map.end()) { in AddAllocationOrIncrementRefCount()211 allocation_map[device_memory.opaque()] = { in AddAllocationOrIncrementRefCount()222 AllocationMap& allocation_map = opaque_to_allocation_map_[device_ordinal]; in DecrementRefCount() local223 auto it = allocation_map.find(device_memory.opaque()); in DecrementRefCount()224 TF_RET_CHECK(it != allocation_map.end()); in DecrementRefCount()229 allocation_map.erase(it); in DecrementRefCount()