Lines Matching refs:request
84 Status EagerServiceImpl::CreateContext(const CreateContextRequest* request,
97 request->server_def().job_name().data(),
98 request->server_def().task_index()),
108 auto* r = env_->rendezvous_mgr->Find(request->rendezvous_id());
109 auto session_name = strings::StrCat("eager_", request->rendezvous_id());
111 session_name, request->server_def(), true));
123 request->async(), std::move(device_mgr), r));
133 new ServerContext(std::move(ctx), request->keep_alive_secs(), env_));
202 Status EagerServiceImpl::Enqueue(const EnqueueRequest* request,
205 TF_RETURN_IF_ERROR(GetServerContext(request->context_id(), &context));
208 for (const auto& item : request->queue()) {
221 Status EagerServiceImpl::WaitQueueDone(const WaitQueueDoneRequest* request,
224 TF_RETURN_IF_ERROR(GetServerContext(request->context_id(), &context));
227 if (request->op_id_size() > 0) {
235 Status EagerServiceImpl::KeepAlive(const KeepAliveRequest* request,
238 TF_RETURN_IF_ERROR(GetServerContext(request->context_id(), &context));
244 Status EagerServiceImpl::CloseContext(const CloseContextRequest* request,
247 if (!GetServerContext(request->context_id(), &context).ok()) {
255 contexts_.erase(request->context_id());
266 const RegisterFunctionRequest* request,
269 TF_RETURN_IF_ERROR(GetServerContext(request->context_id(), &context));
272 return context->Context()->AddFunctionDef(request->function_def());
275 Status EagerServiceImpl::SendTensor(const SendTensorRequest* request,
278 TF_RETURN_IF_ERROR(GetServerContext(request->context_id(), &context));
282 for (const auto& tensor_proto : request->tensors()) {
293 request->device_name().c_str(),
299 context->AddOperationOutputs(tensors, request->op_id());
