Method BestLifetimePolicy

tensorflow/core/common_runtime/gpu_memory_planner.cc:59–70

lifetime policy

// lifetime policy
GPULifetimePolicy* GPUMemoryPlanner::BestLifetimePolicy() {
  GPULifetimePolicy* best_policy = nullptr;
  auto total_mem = std::numeric_limits<size_t>::max();
  for (auto policy : lifetime_stats_polices_) {
    auto policy_mem = policy->TotalMem();
    if (policy_mem < total_mem) {
      best_policy = policy;
      total_mem = policy_mem;
    }
  }
  return best_policy;
}

std::vector<GPULifetimeBin*>& GPUMemoryPlanner::GetSmallBins() {
  return small_bins_;