Lines Matching refs:Rendezvous
48 // to formulate the unique rendezvous key.
69 // proactively cache the rendezvous key for the top-level.
71 OP_REQUIRES_OK(ctx, Rendezvous::ParseKey(parsed_key_.buf_, &parsed_key_));
79 ctx, ctx->rendezvous() != nullptr,
80 errors::Internal("Op kernel context needs to provide a rendezvous."));
86 Rendezvous::Args args;
92 // Use the cached rendezvous key.
94 ctx->SetStatus(ctx->rendezvous()->Send(parsed_key_, args, ctx->input(0),
98 Rendezvous::ParsedKey in_loop_parsed;
102 Rendezvous::ParseKey(in_loop_parsed.buf_, &in_loop_parsed));
104 ctx->SetStatus(ctx->rendezvous()->Send(in_loop_parsed, args, ctx->input(0),
137 // proactively cache the rendezvous key for the top-level.
139 OP_REQUIRES_OK(ctx, Rendezvous::ParseKey(parsed_key_.buf_, &parsed_key_));
146 Rendezvous::DoneCallback make_recv_callback(OpKernelContext* ctx,
152 const Status& s, const Rendezvous::Args& send_args,
153 const Rendezvous::Args& recv_args, const Tensor& val,
173 ctx, ctx->rendezvous() != nullptr,
174 errors::Internal("Op kernel context needs to provide a rendezvous."));
176 Rendezvous::Args args;
183 ctx->rendezvous()->RecvAsync(parsed_key_, args,
186 Rendezvous::ParsedKey in_loop_parsed;
190 ctx, Rendezvous::ParseKey(in_loop_parsed.buf_, &in_loop_parsed), done);
191 ctx->rendezvous()->RecvAsync(in_loop_parsed, args,
