Method AllocateRaw

tensorflow/core/common_runtime/tensorpool_allocator.cc:151–165

}

void* TensorPoolAllocator::AllocateRaw(size_t alignment,
    size_t num_bytes) {
  if (SmallAlloc(num_bytes)) {
    auto header_size = std::max(sizeof(LightHeader), alignment);
    auto total = num_bytes + header_size;
    auto ptr = sub_allocator_->Alloc(alignment, total);
    return SetLightHeader(ptr, total, header_size);
  }

  if (unlikely(stats_)) {
    return BigAllocateStatistic(alignment, num_bytes);
  } else {
    return BigAllocate(alignment, num_bytes);
  }
}

void TensorPoolAllocator::DeallocateRaw(void* ptr) {
  auto light_header = GetLightHeader(ptr);