Method CollectDone

tensorflow/core/common_runtime/memory_planner.cc:125–130

}

void MemoryPlanner::CollectDone() {
  if (allocator_ != nullptr) {
    allocator_->Init();
  }
  Cleanup();
}

void MemoryPlanner::Cleanup() {
  for (auto policy : lifetime_stats_polices_) {