Lines Matching refs:step_id
81 void Worker::AbortStep(int64 step_id) {
82 Rendezvous* rendez = env_->rendezvous_mgr->Find(step_id);
83 SchedNonBlockingClosureAfter(1000000, [rendez, step_id]() {
87 rendez->StartAbort(errors::Aborted("Step ", step_id));
136 const int64 step_id = request->step_id();
137 TRACEPRINTF("RunGraph: %lld", step_id);
156 opts->SetCancelCallback([this, cm, step_id]() {
158 AbortStep(step_id);
176 request->graph_handle(), step_id, session.get(), request->exec_opts(),
178 [this, step_id, response, session, cm, out, token, collector, opts,
181 s = session->graph_mgr->RecvOutputs(step_id, out);
209 const int64 step_id = request->step_id();
211 TRACEPRINTF("PartialRunGraph: %lld", step_id);
229 bool is_new_partial_run = partial_run_mgr_.FindOrCreate(step_id, &cm);
232 opts->SetCancelCallback([this, cm, step_id]() {
234 AbortStep(step_id);
248 graph_handle, step_id, session.get(), request->exec_opts(),
250 [this, token, step_id, session, cm](Status s) {
255 partial_run_mgr_.ExecutorDone(step_id, s);
259 s = session->graph_mgr->SendInputs(step_id, in);
267 step_id, out, [this, out, request, response, step_id, finish](Status s) {
277 partial_run_mgr_.PartialRunDone(step_id, finish, s);
287 const int64 step_id = request->step_id();
288 env_->rendezvous_mgr->Cleanup(step_id);
