Method Reserve

oneflow/core/common/tensor_buffer.cpp:93–105

}

void TensorBufferImpl::Reserve(size_t new_size) {
  if (new_size > buffer_size_) {
    size_t growth_size = std::max(new_size, GetTensorBufferGrowthSize(new_size));
    DeallocateBuffer();
    AllocateBuffer(growth_size);
  } else {
    size_t shrink_size = GetTensorBufferShrinkSize(buffer_size_);
    if (new_size <= shrink_size) {
      DeallocateBuffer();
      AllocateBuffer(shrink_size);
    }
  }
}

void TensorBufferImpl::CopyFrom(const TensorBufferImpl* src) {
  if (src == this) { return; }