Method ClearCaches

tensorflow/core/common_runtime/eager/context.cc:154–172

}

void EagerContext::ClearCaches() {
  {
    mutex_lock ml(executor_map_mu_);
    for (auto& entry : thread_local_executor_) {
      entry.second->WaitForAllPendingNodes().IgnoreError();
    }
  }
  {
    // The executor stores pointers to kernels, so we need to make sure that no
    // async eager ops are still executing. We lock the cache during this time
    // as well.
    mutex_lock ml(cache_mu_);
    default_executor_.WaitForAllPendingNodes().IgnoreError();
    kernel_cache_.clear();
    for (auto& entry : registered_functions_) {
      entry.second->cached_kernel_keys->clear();
    }
  }
}

void EagerContext::SetThreadLocalDevicePlacementPolicy(
    ContextDevicePlacementPolicy policy) {