Method Allocate

tensorflow/core/common_runtime/gpu_tensorpool_allocator.cc:325–331

}

void* GPUTensorPoolAllocator::Bin::Allocate() {
  auto ptr = buffer_.Allocate();
  if (ptr != nullptr) {
    return ptr;
  }
  return virtual_buffer_.Allocate();
}

void* GPUTensorPoolAllocator::Bin::AllocateRaw() {
  return buffer_.Allocate();