Method AllocateRaw

tensorflow/core/common_runtime/gpu_tensorpool_allocator.cc:203–219

}

void* GPUTensorPoolAllocator::AllocateRaw(size_t alignment,
    size_t num_bytes) {
  if (!inited_.load()) {
    auto ptr = sub_allocator_->Alloc(alignment, num_bytes);
    mem_planner_->TrackAllocate(alignment, num_bytes, ptr);
    return ptr;
  }

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

void GPUTensorPoolAllocator::DeallocateRaw(void* ptr) {
  if (!inited_.load()) {