Lines Matching refs:step_id
44 // until the tensor is received. Each global unique "step_id"
67 // "step_id". The caller takes ownership of one reference on the
72 RemoteRendezvous* Find(int64 step_id) override;
74 // Finds the local rendezvous instance for the "step_id". Runs
78 void RecvLocalAsync(int64 step_id, const Rendezvous::ParsedKey& parsed,
82 Status RecvLocal(int64 step_id, const Rendezvous::ParsedKey& parsed,
85 // Removes rendezvous for "step_id".
89 void Cleanup(int64 step_id) override;
95 virtual BaseRemoteRendezvous* Create(int64 step_id,
99 // Maps step_id to rendezvous.
108 BaseRemoteRendezvous* FindOrCreate(int64 step_id);
121 BaseRemoteRendezvous(const WorkerEnv* env, int64 step_id);
