MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / CleanupGraphAsync

Method CleanupGraphAsync

tensorflow/core/distributed_runtime/worker.cc:367–382  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

365}
366
367void Worker::CleanupGraphAsync(const CleanupGraphRequest* request,
368 CleanupGraphResponse* response,
369 StatusCallback done) {
370 const int64 step_id = request->step_id();
371 env_->rendezvous_mgr->Cleanup(step_id);
372 if (env_->collective_executor_mgr) {
373 env_->collective_executor_mgr->Cleanup(step_id);
374 }
375 for (Device* d : env_->local_devices) {
376 ScopedAllocatorMgr* sam = d->GetScopedAllocatorMgr();
377 if (sam) {
378 sam->Cleanup(step_id);
379 }
380 }
381 done(Status::OK());
382}
383
384void Worker::CleanupAllAsync(const CleanupAllRequest* request,
385 CleanupAllResponse* response,

Callers 2

CleanUpMethod · 0.45

Calls 3

step_idMethod · 0.45
CleanupMethod · 0.45
GetScopedAllocatorMgrMethod · 0.45

Tested by

no test coverage detected