Method Buffer

tensorflow/core/common_runtime/gpu_tensorpool_allocator.cc:345–359

}

GPUTensorPoolAllocator::Buffer::Buffer(size_t len, size_t chunk_size,
    size_t alignment, void* begin) {
  auto rounded_bytes = RoundedBytes(chunk_size, alignment);
  auto buffer_size = rounded_bytes * len;
  begin_ = begin;
  if (begin != nullptr) {
    end_ = static_cast<char*>(begin) + buffer_size;
  } else {
    end_ = nullptr;
  }

  for (auto i = 0; i < len; ++i) {
    buffer_.emplace(static_cast<char*>(begin) + rounded_bytes *i);
  }
}

void* GPUTensorPoolAllocator::Buffer::Allocate() {
  std::lock_guard<spin_lock> l(lock_);