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

Method AbortStep

tensorflow/core/distributed_runtime/worker.cc:115–125  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

113}
114
115void Worker::AbortStep(int64 step_id) {
116 Rendezvous* rendez = env_->rendezvous_mgr->Find(step_id);
117 SchedNonBlockingClosureAfter(1000000, [rendez, step_id]() {
118 // Delay a bit before aborting the step. This way, the root
119 // cause may return first back to the client instead of this
120 // cancellation generated abort error.
121 rendez->StartAbort(errors::Aborted("Step ", step_id,
122 " cancelled. Cancelling rendezvous."));
123 rendez->Unref();
124 });
125}
126
127Status Worker::PrepareRunGraph(RunGraphRequestWrapper* req,
128 GraphMgr::NamedTensors* in,

Callers

nothing calls this directly

Calls 4

FindMethod · 0.45
StartAbortMethod · 0.45
UnrefMethod · 0.45

Tested by

no test coverage detected