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

Method DeregisterGraphAsync

tensorflow/core/distributed_runtime/worker.cc:97–113  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

95}
96
97void Worker::DeregisterGraphAsync(const DeregisterGraphRequest* request,
98 DeregisterGraphResponse* response,
99 StatusCallback done) {
100 std::shared_ptr<WorkerSession> session;
101 Status s;
102 if (request->create_worker_session_called()) {
103 s = env_->session_mgr->WorkerSessionForSession(request->session_handle(),
104 &session);
105 } else {
106 session = env_->session_mgr->LegacySession();
107 }
108 if (s.ok()) {
109 s = session->graph_mgr->Deregister(request->graph_handle());
110 }
111
112 done(s);
113}
114
115void Worker::AbortStep(int64 step_id) {
116 Rendezvous* rendez = env_->rendezvous_mgr->Find(step_id);

Callers 1

DeregisterPartitionsMethod · 0.45

Calls 6

session_handleMethod · 0.80
LegacySessionMethod · 0.45
okMethod · 0.45
DeregisterMethod · 0.45

Tested by

no test coverage detected