Method AllocatedSize

tensorflow/core/framework/tracking_allocator.cc:213–224

}

size_t TrackingAllocator::AllocatedSize(const void* ptr) const {
  if (track_sizes_locally_) {
    mutex_lock lock(mu_);
    auto it = in_use_.find(ptr);
    if (it != in_use_.end()) {
      return (*it).second.allocated_size;
    }
    return 0;
  } else {
    return allocator_->AllocatedSize(ptr);
  }
}

int64 TrackingAllocator::AllocationId(const void* ptr) const {
  if (track_sizes_locally_) {