Method TensorBufferPool

oneflow/core/common/tensor_buffer.cpp:212–218

}  // namespace

TensorBufferPool::TensorBufferPool()
    : thread_local_cache_size_(GetTensorBufferPoolThreadLocalCacheSize()),
      pool_size_(GetTensorBufferPoolSize()) {
  auto& thread_local_cache = ThreadLocalCache();
  thread_local_cache.reserve(thread_local_cache_size_);
  global_free_list_.reserve(pool_size_);
}

void TensorBufferPool::Allocate(ItemT* item, const Shape& shape, DataType dtype) {
  CHECK(!(*item)) << "TensorBuffer is already allocated";