Method BeginStep

tensorflow/core/common_runtime/gpu_tensorpool_allocator.cc:185–201

}

void GPUTensorPoolAllocator::BeginStep() {
  if (inited_.load()) {
    for (auto b : lifetime_bins_) {
      if (b != nullptr) {
        b->BeginStep();
      }
    }
    for (auto it : large_lifetime_bins_) {
      it.second->BeginStep();
    }
  }
  std::lock_guard<spin_lock> l(free_lock_);
  for (auto ptr : async_free_list_) {
    sub_allocator_->Free(ptr, 0);
  }
  async_free_list_.clear();
}

void* GPUTensorPoolAllocator::AllocateRaw(size_t alignment,
    size_t num_bytes) {